commit main

main.py

@@ -37,9 +37,12 @@ app.add_middleware(
 from transformers import VisionEncoderDecoderModel, ViTImageProcessor, AutoTokenizer
 try:
     interpreter =None
+    print("installing interpreter ...")
     interpreter =pipeline("image-to-text", model="Salesforce/blip-image-captioning-base")
     if interpreter is None :
         print("\n\n interpreter is nonne \n\n")
+    else:
+        print(" interpreter installed.")
     #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")
@@ -51,9 +54,12 @@ except Exception as exp:
 
 try:
     summarizer=None
+    print ("installing summarizer ...")
     summarizer = pipeline("summarization", model="facebook/bart-large-cnn")
     if summarizer is None :
         print("\n\n summarizer is nonne \n\n")
+    else:
+        print(" summarizer installed.")
 except Exception as exp:
     print("[ERROR] Can't load facebook/bart-large-cnn ")
     print(str(exp))
@@ -70,9 +76,12 @@ except Exception as exp:
 
 try:
     generator=None
+    print("installing generator ...")
     generator = pipeline("text-generation", model="deepseek-ai/deepseek-coder-1.3b-instruct")
     if generator is None :
         print("\n\n generator is nonne \n\n")
+    else:
+        print(" generator installed.")
 except Exception as exp:
     print("[ERROR] Can't load deepseek-ai/deepseek-coder-1.3b-instruct ")
     print(str(exp))