Add application file

app.py

@@ -31,20 +31,24 @@ augment = transforms.Compose([
 
 # FONCTION PRINCIPALE
 def generate_recipe(image: Image.Image):
-    yield "🔄 Traitement de l'image... Veuillez patienter."
+    try:
+        yield "🔄 Traitement de l'image..."
 
-    # Augmentation
-    image_aug = augment(image)
+        image_aug = image
 
-    # Classification
-    results = ingredient_classifier(image_aug)
-    ingredients = [res["label"] for res in results]
-    ingredient_str = ", ".join(ingredients)
+        yield "📸 Classification en cours..."
+        results = ingredient_classifier(image_aug)
+        ingredients = [res["label"] for res in results]
+        ingredient_str = ", ".join(ingredients)
 
-    yield f"🥕 Ingrédients détectés : {ingredient_str}\n\n🍳 Génération de la recette..."
-    prompt = f"Ingredients: {ingredient_str}. Recipe:"
-    recipe = recipe_generator(prompt, max_new_tokens=256, do_sample=True)[0]["generated_text"]
-    yield f"### 🥕 Ingrédients détectés :\n{ingredient_str}\n\n### 🍽️ Recette générée :\n{recipe}"
+        yield f"🥕 Ingrédients détectés : {ingredient_str}\n\n🍳 Génération de la recette..."
+        prompt = f"Ingredients: {ingredient_str}. Recipe:"
+        recipe = recipe_generator(prompt, max_new_tokens=256, do_sample=True)[0]["generated_text"]
+
+        yield f"### 🥕 Ingrédients détectés :\n{ingredient_str}\n\n### 🍽️ Recette générée :\n{recipe}"
+
+    except Exception as e:
+        yield f"❌ Une erreur est survenue : {str(e)}"
 
 # INTERFACE
 interface = gr.Interface(

requirements.txt

@@ -1,5 +1,5 @@
-gradio
-transformers
-torch
-torchvision
-pillow
+gradio==4.25.0
+transformers==4.40.1
+torch>=2.1.0
+torchvision>=0.16.0
+pillow>=10.0.0