Update app.py

app.py

@@ -20,27 +20,28 @@ if not os.path.exists(UPLOAD_FOLDER):
     os.makedirs(UPLOAD_FOLDER)
 
 # Load PaliGemma model and processor (load once)
+import torch
+from transformers import PaliGemmaForConditionalGeneration, AutoProcessor
+
 def load_paligemma_model():
     try:
-        print("Loading PaliGemma model from local path...")
+        print("Loading PaliGemma model from Hugging Face...")
         device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
         print(f"Using device: {device}")
 
-        # Specify the local path relative to your project structure
-        local_model_path = os.path.join(os.path.dirname(__file__), 'Model')  # Update this path
-
-        # Load model and processor from the specified local path
+        # Load model and processor from Hugging Face Hub
+        model_name = "ahmed-masry/chartgemma"  # Update with the correct model name if needed
         model = PaliGemmaForConditionalGeneration.from_pretrained(
-            local_model_path,
+            model_name,
             torch_dtype=torch.float16
         )
-        processor = AutoProcessor.from_pretrained(local_model_path)
+        processor = AutoProcessor.from_pretrained(model_name)
         model = model.to(device)
+
         print("Model loaded successfully")
         return model, processor, device
     except Exception as e:
         print(f"Error loading model: {str(e)}")
-        traceback.print_exc()
         raise