Update app.py

app.py

@@ -1,6 +1,7 @@
 from flask import Flask, request, jsonify
 from transformers import AutoProcessor, CLIPModel
 from PIL import Image
+import base64
 
 # Charger le modèle CLIP et le processeur
 model = CLIPModel.from_pretrained("patrickjohncyh/fashion-clip")
@@ -15,6 +16,7 @@ app = Flask(__name__)
 def classify_image_with_text(text, image):
     # Effectuer la classification d'image à l'aide du texte
     keywords = text.split(',')
+    image = decode_image_from_base64(image)
     inputs = processor(
             text=keywords, images=image, return_tensors="pt", padding=True
     )
@@ -25,6 +27,11 @@ def classify_image_with_text(text, image):
     predicted_label = keywords[predicted_class_index]
     return predicted_label
 
+# Fonction pour décoder une image encodée en base64 en objet PIL.Image.Image
+def decode_image_from_base64(image_data):
+    image_data = base64.b64decode(image_data)
+    image = Image.open(io.BytesIO(image_data))
+    return image
 
 @app.get("/")
 def root():