Update app.py

app.py

@@ -10,9 +10,9 @@ def query(payload):
     response = requests.post(API_URL, headers=headers, json=payload)
     return response.json()
 
-def generate_yes_no_response(question):
+def reponse(question):
     # Créez un prompt qui force le modèle à choisir entre Oui et Non
-    prompt = f" {question}\nanswer only yes or no :"
+    prompt = {question}
     
    # Configurez les paramètres pour forcer une réponse courte
     payload = {
@@ -28,19 +28,9 @@ def generate_yes_no_response(question):
     
     response = query(payload)
     
-    # Assurez-vous que la réponse est soit "Oui" soit "Non"
-    if isinstance(response, list) and len(response) > 0:
-        answer = response[0].get('generated_text', '').strip().lower()
-        if answer == "yes":
-            return "yes"
-        elif answer == "no":
-            return "no"
-    
-    # Si la réponse n'est pas "Oui" ou "Non", retournez une réponse par défaut
-    return "Réponse invalide"
-
+  
 def chatbot(message, history):
-    response = generate_yes_no_response(message)
+    response = reponse(message)
     return response
 
 iface = gr.ChatInterface(