Correct trash

app.py

@@ -4,43 +4,62 @@ from PIL import Image
 import torch
 from torchvision import transforms
 
+# MODELES
 INGREDIENT_MODEL_ID = "stchakman/Fridge_Items_Model"
+RECIPE_MODEL_ID = "flax-community/t5-recipe-generation"
+
+# PIPELINES
 ingredient_classifier = pipeline(
     "image-classification",
     model=INGREDIENT_MODEL_ID,
     device=0 if torch.cuda.is_available() else -1,
-    top_k=5 
+    top_k=5
 )
 
-RECIPE_MODEL_ID = "flax-community/t5-recipe-generation"
-recipe_generator = pipeline("text2text-generation", model=RECIPE_MODEL_ID)
+recipe_generator = pipeline(
+    "text2text-generation",
+    model=RECIPE_MODEL_ID,
+    device=0 if torch.cuda.is_available() else -1
+)
 
+# AUGMENTATION
 augment = transforms.Compose([
     transforms.RandomHorizontalFlip(p=0.5),
     transforms.RandomRotation(10),
     transforms.ColorJitter(brightness=0.2, contrast=0.2),
 ])
 
+# FONCTION PRINCIPALE
 def generate_recipe(image: Image.Image):
+    # Affichage "Processing..." temporaire
+    yield "🔄 Traitement de l'image... Veuillez patienter."
+
+    # Augmentation
     image_aug = augment(image)
+
+    # Classification
     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..."
+
+    # Génération recette
     prompt = f"Ingredients: {ingredient_str}. Recipe:"
-    recipe = recipe_generator(prompt, max_length=300, do_sample=True)[0]["generated_text"]
-    
-    return f"### Ingredients detectes :\n{ingredient_str}\n\n### Recette generee :\n{recipe}"
+    recipe = recipe_generator(prompt, max_new_tokens=256, do_sample=True)[0]["generated_text"]
+
+    # Affichage final
+    yield f"### 🥕 Ingrédients détectés :\n{ingredient_str}\n\n### 🍽️ Recette générée :\n{recipe}"
 
+# INTERFACE
 interface = gr.Interface(
     fn=generate_recipe,
-    inputs=gr.Image(type="pil"),
+    inputs=gr.Image(type="pil", label="📷 Image de vos ingrédients"),
     outputs=gr.Markdown(),
-    title="🥕 Generateur de recettes 🧑‍🍳",
-    description="Envoyer une image d'ingredients pour recevoir une recette",
+    title="🥕 Générateur de Recettes 🧑‍🍳",
+    description="Dépose une image d'ingrédients pour obtenir une recette automatiquement générée à partir d'un modèle IA.",
     allow_flagging="never"
 )
 
 if __name__ == "__main__":
-    interface.launch()
+    interface.launch(share=True)  # Mets share=True pour générer le lien Gradio public