Update scripts/visual_qa.py

scripts/visual_qa.py

@@ -17,8 +17,17 @@ from smolagents import Tool, tool
 
 load_dotenv(override=True)
 
-idefics_processor = AutoProcessor.from_pretrained("HuggingFaceM4/idefics2-8b-chatty")
-
+#idefics_processor = AutoProcessor.from_pretrained("HuggingFaceM4/idefics2-8b-chatty")
+# Load models with increased timeout and retries
+try:
+    idefics_processor = AutoProcessor.from_pretrained(
+        "HuggingFaceM4/idefics2-8b-chatty",
+        timeout=60,  # Double timeout
+        use_auth_token=os.getenv("HF_TOKEN")
+    )
+except Exception as e:
+    warnings.warn(f"Couldn't load Idefics processor: {str(e)}")
+    idefics_processor = None
 
 def process_images_and_text(image_path, query, client):
     messages = [