Update app.py

app.py

@@ -16,11 +16,12 @@ 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)  
-    print(probs)
-     # Récupérer l'index de la prédiction avec le score le plus élevé
+    # Récupérer l'index de la prédiction avec le score le plus élevé
     predicted_class_index = probs.argmax(dim=1).item()
+    # Récupérer le tokenizer associé au processeur
+    tokenizer = processor.tokenizer
     # Récupérer le label correspondant à l'index
-    label = processor.tokenizer.cls_token + processor.convert_ids_to_tokens(predicted_class_index)
+    label = tokenizer.convert_ids_to_tokens(predicted_class_index)
     return label
     
 # Créer l'interface Gradio avec l'API de Gradio Blocks