Update app.py

app.py

@@ -10,14 +10,15 @@ processor = AutoProcessor.from_pretrained("patrickjohncyh/fashion-clip")
 # Définir la fonction pour la classification d'image avec du texte en entrée
 def classify_image_with_text(text, image):
     # Effectuer la classification d'image à l'aide du texte
+    keywords = text.split(',')
     inputs = processor(
-            text=text, images=image, return_tensors="pt", padding=True
+            text=keywords, images=image, return_tensors="pt", padding=True
     )
     outputs = model(**inputs)
     logits_per_image = outputs.logits_per_image  # this is the image-text similarity score
     probs = logits_per_image.softmax(dim=1) 
     predicted_class_index = probs.argmax(dim=1).item()
-    predicted_label = text[predicted_class_index]
+    predicted_label = keywords[predicted_class_index]
     return predicted_class_index
     
 # Créer l'interface Gradio avec l'API de Gradio Blocks