update main

main.py

@@ -18,6 +18,8 @@ from pptx import Presentation
 import seaborn as sns
 import PIL.Image as Image
 
+
+
 import fitz
 
 
@@ -38,20 +40,22 @@ try:
 except Exception as exp:
     print("[ERROR] Can't load nlpconnect/vit-gpt2-image-captioning")
     print(str(exp))
-    
-#try:
-#    summarizer = pipeline("summarization", model="facebook/bart-large-cnn",device=0)
-#except Exception as exp:
-#    print("[ERROR] Can't load facebook/bart-large-cnn ")
-#    print(str(exp))
+
+
 
 try:
-    summarizer_model = BartForConditionalGeneration.from_pretrained("facebook/bart-large-cnn")
-except OSError as e:
-    print(f"[INFO] PyTorch weights not found. Falling back to TensorFlow weights.\n{e}")
-    summarizer_model = BartForConditionalGeneration.from_pretrained("facebook/bart-large-cnn", from_tf=True)
+    summarizer = pipeline("summarization", model="facebook/bart-large-cnn")
+except Exception as exp:
+    print("[ERROR] Can't load facebook/bart-large-cnn ")
+    print(str(exp))
+
+#try:
+#    summarizer_model = BartForConditionalGeneration.from_pretrained("facebook/bart-large-cnn")
+#except OSError as e:
+#    print(f"[INFO] PyTorch weights not found. Falling back to TensorFlow weights.\n{e}")
+#    summarizer_model = BartForConditionalGeneration.from_pretrained("facebook/bart-large-cnn", from_tf=True)
 
-summarizer_tokenizer = BartTokenizer.from_pretrained("facebook/bart-large-cnn")
+#summarizer_tokenizer = BartTokenizer.from_pretrained("facebook/bart-large-cnn")