Update app.py

app.py

@@ -3,23 +3,23 @@ import google.generativeai as genai
 from yolov5 import YOLOv5
 from PIL import Image
 
-# Configurez votre clé API
+# Clé API
 genai.configure(api_key="AIzaSyB_Hnl_77gd1I8xs6iuLqKIoGHgsZMXm1M")
 
-# Chargez le modèle Gemini Pro
+# Le modèle Gemini Pro
 gemini_model = genai.GenerativeModel('gemini-pro')
 
-# Charger le modèle de détection YOLOv5
+# Le modèle de détection YOLOv5
 yolo_model_path = "yolov5s.pt"
 yolo_model = YOLOv5(yolo_model_path, device="cpu")
 
-# Fonction de détection d'ingrédients
+# Détection d'ingrédients
 def detect_ingredients(image):
     results = yolo_model.predict(image)
     ingredients = results.pandas().xyxy[0]['name'].tolist()
     return ingredients
 
-# Fonction de génération de recettes
+# Prompt de génération de recettes (Mistral AI)
 def generate_recipe(ingredients):
     # Ajoutez une indication dans le prompt pour générer la réponse en français
     prompt = f"""
@@ -55,8 +55,6 @@ def generate_recipe(ingredients):
     return response.text
 
 
-
-
 # Interface Gradio
 def process_image(image):
     # Ouvrir l'image depuis le chemin fourni
@@ -74,7 +72,7 @@ iface = gr.Interface(
     description="Téléchargez une image d'ingrédients pour générer une recette.",
 )
 
-# Lancer l'application
+# Pour lancer l'application
 if __name__ == "__main__":
     iface.launch()