Update app.py

app.py

@@ -15,8 +15,9 @@ def classify_image_with_text(text, image):
     )
     outputs = model(**inputs)
     logits_per_image = outputs.logits_per_image  # this is the image-text similarity score
-    probs = logits_per_image.softmax(dim=1)  
-    return probs
+    probs = logits_per_image.softmax(dim=1) 
+    predicted_class_index = probs.argmax(dim=1).item()
+    return predicted_class_index
     
 # Créer l'interface Gradio avec l'API de Gradio Blocks
 with gr.Interface(