Update app.py

app.py

@@ -13,11 +13,11 @@ def classify_image_with_text(text, image):
             text=["a photo of a man", "a photo of a woman"], images=image, return_tensors="pt", padding=True
     )
     outputs = model(**inputs)
-    # Récupérer les scores et les libellés
-    scores = outputs.logits_per_image
-    predictions = outputs.predictions
-    labels = [processor.tokenizer.decode([p]) for p in predictions]
-    return labels, scores
+    # Récupérer l'index de l'étiquette avec le score le plus élevé
+    predicted_class_index = outputs.logits_per_image.argmax(dim=-1)
+    # Convertir l'index de l'étiquette en libellé
+    label = processor.convert_ids_to_tokens(predicted_class_index.item())
+    return label
     
 # Créer l'interface Gradio avec l'API de Gradio Blocks
 with gr.Interface(