Update app.py

app.py

@@ -73,8 +73,25 @@ try:
     print("Model loaded successfully!")
     
     print("Loading processor...")
-    processor = AutoProcessor.from_pretrained(MODEL_PATH, local_files_only=True)
-    print("Processor loaded successfully!")
+    try:
+        processor = AutoProcessor.from_pretrained(MODEL_PATH, local_files_only=True)
+        print("Processor loaded successfully!")
+    except Exception as proc_error:
+        print(f"Error loading processor from local files: {proc_error}")
+        print("Attempting to load processor from base SigLIP model...")
+        
+        # Try to load processor from the base SigLIP model
+        try:
+            processor = AutoProcessor.from_pretrained("google/siglip-base-patch16-224")
+            print("Processor loaded from base SigLIP model!")
+        except Exception as base_error:
+            print(f"Error loading base processor: {base_error}")
+            print("Trying alternative processor...")
+            
+            # As a last resort, try to create a minimal processor
+            from transformers import CLIPImageProcessor
+            processor = CLIPImageProcessor.from_pretrained("openai/clip-vit-base-patch32")
+            print("Using CLIP processor as fallback!")
     
     # Get labels - handle case where id2label might not exist
     if hasattr(model.config, 'id2label') and model.config.id2label: