Update app.py

app.py

@@ -12,8 +12,11 @@ def classify_image_with_text(text, image):
     inputs = processor(
             text=[text], images=image, return_tensors="pt", padding=True
     )
-    return model(**inputs)
-
+    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
+    
 # Créer l'interface Gradio avec l'API de Gradio Blocks
 with gr.Interface(
     fn=classify_image_with_text,