Update app.py

app.py

@@ -12,9 +12,9 @@ def query(payload):
 
 def generate_yes_no_response(question):
     # Créez un prompt qui force le modèle à choisir entre Oui et Non
-    prompt = f"Question : {question}\nRéponse (Oui ou Non) :"
+    prompt = f" {question}\nanswer only yes or no :"
     
-    # Configurez les paramètres pour forcer une réponse courte
+   # Configurez les paramètres pour forcer une réponse courte
     payload = {
         "inputs": prompt,
         "parameters": {
@@ -31,10 +31,10 @@ def generate_yes_no_response(question):
     # 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 == "oui":
-            return "Oui"
-        elif answer == "non":
-            return "Non"
+        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"