Update app.py

app.py

@@ -19,7 +19,7 @@ def classify_image_with_text(text, image):
     probs = logits_per_image.softmax(dim=1) 
     predicted_class_index = probs.argmax(dim=1).item()
     predicted_label = keywords[predicted_class_index]
-    return predicted_class_index
+    return predicted_label
     
 # Créer l'interface Gradio avec l'API de Gradio Blocks
 with gr.Interface(