test

app.py

@@ -1,16 +1,15 @@
-from click import prompt
 import gradio as gr
 
-# Charger le modèle OpenChat
+# Charger le modèle OpenChat en dehors de la fonction pour éviter de le charger à chaque appel
 openchat_model = gr.load("models/openchat/openchat_3.5")
 
 # Définir une fonction de génération de poème en fonction des entrées
 def generate_poem(author, style, syllables, activities):
-    # Convertir la chaîne d'activités en liste
-    activities_list = activities.split(", ") if activities else []
+    # Construire le prompt de manière plus explicite
+    prompt_text = f"Author: {author}, Style: {style}, Syllables: {syllables}"
 
-    # Construire le prompt
-    prompt_text = f"Author: {author}, Style: {style}, Syllables: {syllables}, Activities: {activities_list}"
+    if activities:
+        prompt_text += f", Activities: {activities}"
 
     # Appeler le modèle OpenChat avec le prompt
     response = openchat_model(prompt_text)