Update app.py

app.py

@@ -2,7 +2,7 @@ import gradio as gr
 import google.generativeai as genai
 from yolov5 import YOLOv5
 from PIL import Image
-
+from transformers import AutoProcessor, BarkModel
 
 #Clé Api
 genai.configure(api_key="AIzaSyB_Hnl_77gd1I8xs6iuLqKIoGHgsZMXm1M")
@@ -15,6 +15,10 @@ gemini_model = genai.GenerativeModel('gemini-pro')
 yolo_model_path = "yolov5s.pt"
 yolo_model = YOLOv5(yolo_model_path, device="cpu")
 
+# Modèle Bark
+processor = AutoProcessor.from_pretrained("suno/bark")
+model = BarkModel.from_pretrained("suno/bark")
+
 
 # Détection d'ingrédients
 def detect_ingredients(image):
@@ -50,8 +54,6 @@ def generate_recipe(ingredients):
     response = gemini_model.generate_content(prompt)
     return response.text
 
-    response = gemini_model.generatecontent(prompt)
-    return response.text
 
 
 # Interface Gradio
@@ -61,6 +63,11 @@ def process_image(image):
         recipe = generate_recipe(ingredients)
         return f"Ingrédients détectés : {', '.join(ingredients)}\n\nRecette générée :\n{recipe}"
 
+# Fonction pour générer de l'audio avec le modèle Bark
+def generate_audio(text):
+    inputs = processor(text=text, return_tensors="pt")
+    audio_output = model.generate(**inputs)
+    return audio_output
 
 # Mise à jour de l'interface Gradio
 iface = gr.Interface(