Update app.py

app.py

@@ -78,20 +78,30 @@ try:
         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...")
+        print("Attempting to load just the image processor...")
         
-        # Try to load processor from the base SigLIP model
+        # Try to load just the image processor from your 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...")
+            from transformers import SiglipImageProcessor
+            processor = SiglipImageProcessor.from_pretrained(MODEL_PATH, local_files_only=True)
+            print("Image processor loaded successfully from local files!")
+        except Exception as img_proc_error:
+            print(f"Error loading local image processor: {img_proc_error}")
+            print("Loading image processor from base SigLIP model...")
             
-            # 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!")
+            # Try to load processor from the base SigLIP model
+            try:
+                from transformers import SiglipImageProcessor
+                processor = SiglipImageProcessor.from_pretrained("google/siglip-base-patch16-224")
+                print("Image processor loaded from base SigLIP model!")
+            except Exception as base_error:
+                print(f"Error loading base processor: {base_error}")
+                print("Using CLIP processor as fallback...")
+                
+                # 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: