Add better model

app.py

@@ -4,44 +4,35 @@ from PIL import Image
 import torch
 import re
 
-# —————————————————————————————
-# 1. CLASSIFICATION D’INGReDIENTS
-# —————————————————————————————
-INGREDIENT_MODEL_ID = "stchakman/Fridge_Items_Model"
+# 1. Pipelines
 ingredient_classifier = pipeline(
     "image-classification",
-    model=INGREDIENT_MODEL_ID,
+    model="stchakman/Fridge_Items_Model",
     device=0 if torch.cuda.is_available() else -1,
     top_k=4
 )
 
-# —————————————————————————————
-# 2. MODeLE DE GeNeRATION
-# —————————————————————————————
 recipe_generator = pipeline(
     "text2text-generation",
     model="flax-community/t5-recipe-generation",
     device=0 if torch.cuda.is_available() else -1
 )
 
-# —————————————————————————————
-# 3. FONCTION PRINCIPALE
-# —————————————————————————————
+# 2. Fonction bien définie en amont
 def generate_recipe(image: Image.Image):
     try:
-        # Detection ingredients
         results = ingredient_classifier(image)
         ingredients = [res["label"].lower() for res in results]
+
         if not ingredients:
             return {
-                "error": "Aucun ingredient detecte.",
+                "error": "Aucun ingrédient détecté.",
                 "ingredients_detected": []
             }
 
         prompt = ", ".join(ingredients)
         generated = recipe_generator(prompt, max_length=200)[0]["generated_text"]
 
-        # extraction par expressions regulieres
         title_match = re.search(r"title\s*:\s*(.+?)(ingredients\s*:|$)", generated, re.IGNORECASE | re.DOTALL)
         ingredients_match = re.search(r"ingredients\s*:\s*(.+?)(directions\s*:|$)", generated, re.IGNORECASE | re.DOTALL)
         directions_match = re.search(r"directions\s*:\s*(.+)", generated, re.IGNORECASE | re.DOTALL)
@@ -51,7 +42,6 @@ def generate_recipe(image: Image.Image):
         directions_raw = directions_match.group(1).strip() if directions_match else None
         directions_list = re.split(r"\.\s*", directions_raw) if directions_raw else []
 
-        # nettoyage des listes
         ingredients_list = [line.strip("- ").strip() for line in ingredients_list if line.strip()]
         directions_list = [step.strip("- ").strip() for step in directions_list if step.strip()]
 
@@ -63,21 +53,19 @@ def generate_recipe(image: Image.Image):
                 "instructions": directions_list
             }
         }
-
     except Exception as e:
         return {"error": str(e)}
 
-# —————————————————————————————
-# 4. INTERFACE GRADIO
-# —————————————————————————————
+# 3. Interface Gradio (PAS de `share=True` sur Hugging Face !)
 interface = gr.Interface(
     fn=generate_recipe,
-    inputs=gr.Image(type="pil", label="📷 Image de vos ingredients"),
+    inputs=gr.Image(type="pil", label="📷 Image de vos ingrédients"),
     outputs=gr.JSON(),
-    title="🥕 Generateur de Recettes",
-    description="Deposez une image d'ingredients pour generer une recette automatiquement (resultat JSON structure).",
+    title="🥕 Générateur de Recettes",
+    description="Dépose une image d'ingrédients pour générer une recette automatiquement.",
     allow_flagging="never"
 )
 
+# 4. Lancement
 if __name__ == "__main__":
-    interface.launch(share=True)
+    interface.launch()