add model caption

main.py

@@ -32,11 +32,13 @@ app.add_middleware(
     allow_methods=["*"],
     allow_headers=["*"],
 )
+
+from transformers import VisionEncoderDecoderModel, ViTImageProcessor, AutoTokenizer
 try:
-    interpreter = pipeline("image-to-text", model="nlpconnect/vit-gpt2-image-captioning")
-    #interpreter_model = VisionEncoderDecoderModel.from_pretrained("nlpconnect/vit-gpt2-image-captioning")
-    #interpreter_processor = ViTImageProcessor.from_pretrained("nlpconnect/vit-gpt2-image-captioning")
-    #interpreter_tokenizer = AutoTokenizer.from_pretrained("nlpconnect/vit-gpt2-image-captioning")
+    interpreter =1 #pipeline("image-to-text", model="nlpconnect/vit-gpt2-image-captioning")
+    interpreter_model = VisionEncoderDecoderModel.from_pretrained("nlpconnect/vit-gpt2-image-captioning")
+    interpreter_processor = ViTImageProcessor.from_pretrained("nlpconnect/vit-gpt2-image-captioning")
+    interpreter_tokenizer = AutoTokenizer.from_pretrained("nlpconnect/vit-gpt2-image-captioning")
 except Exception as exp:
     print("[ERROR] Can't load nlpconnect/vit-gpt2-image-captioning")
     print(str(exp))