Update app.py

app.py

@@ -3,10 +3,10 @@ from PIL import Image
 from transformers import AutoProcessor, AutoModelForImageClassification
 import gradio as gr
 
-# Load model and processor from local "model" folder
-model = AutoModelForImageClassification.from_pretrained("model")
-processor = AutoProcessor.from_pretrained("model",use_fast = False)
-labels = model.config.id2label  # e.g., {0: "non-hateful", 1: "hateful"}
+# Load model and processor from Hugging Face
+model = AutoModelForImageClassification.from_pretrained("google/siglip2-base-patch16-naflex")
+processor = AutoProcessor.from_pretrained("google/siglip2-base-patch16-naflex")
+labels = model.config.id2label
 
 def classify_meme(image: Image.Image):
     inputs = processor(images=image, return_tensors="pt").to(model.device)
@@ -27,3 +27,4 @@ demo = gr.Interface(
 
 if __name__ == "__main__":
     demo.launch()
+