Update app.py

app.py

@@ -5,8 +5,11 @@ import gradio as gr
 import pytesseract
 
 # Load model and processor
-model = AutoModelForImageClassification.from_pretrained("google/siglip2-base-patch16-naflex")
-processor = AutoProcessor.from_pretrained("google/siglip2-base-patch16-naflex")
+model = AutoModelForImageClassification.from_pretrained("Chanlefe/SigLIP2_77")
+
+processor = AutoProcessor.from_pretrained("Chanlefe/SigLIP2_77")
+
+
 labels = model.config.id2label
 
 # Classify meme and extract text
@@ -25,7 +28,8 @@ def classify_meme(image: Image.Image):
         "Predictions": predictions,
         "Extracted Text": extracted_text.strip()
     }
-
+print("Extracted Text:", extracted_text)
+print("Predictions:", predictions)
 # Gradio interface
 demo = gr.Interface(
     fn=classify_meme,