Update app.py

app.py

@@ -1,5 +1,4 @@
-from fastapi import FastAPI, Form, Response, UploadFile, File
-from fastapi.middleware.cors import CORSMiddleware
+from flask import Flask, request, jsonify
 from transformers import AutoProcessor, CLIPModel
 from PIL import Image
 
@@ -9,17 +8,8 @@ processor = AutoProcessor.from_pretrained("patrickjohncyh/fashion-clip")
 
 
 # Créer une instance FastAPI
-app = FastAPI()
+app = Flask(__name__)
 
-origins = ["*"]
-
-app.add_middleware(
-    CORSMiddleware,
-    allow_origins=origins,
-    allow_credentials=True,
-    allow_methods=["*"],
-    allow_headers=["*"],
-)
 
 # Définir la fonction pour la classification d'image avec du texte en entrée
 def classify_image_with_text(text, image):
@@ -40,12 +30,15 @@ def classify_image_with_text(text, image):
 async def root():
     return {"message": "Welcome to the Fashion Clip API!"}
 
-# Définir une route d'API
-@app.post("/classify")
-async def classify(prompt: str, image: UploadFile = None):
-    image_bytes = await image.read()
-    image_pil = Image.open(io.BytesIO(image_bytes))
-    return {"predicted_label": classify_image_with_text(prompt, image_pil)}
+# Route pour l'API REST
+@app.route('/api/classify', methods=['POST'])
+def classify():
+    data = request.json
+    text = data['text']
+    image = data['image']
+    result = classify_image_with_text(text, image)
+    return jsonify({'result': result})
+
 if __name__ == '__main__':
     app.run(debug=True)