Update app.py

app.py

@@ -1,9 +1,7 @@
 import gradio as gr
 import requests
 import os
-import random
 
-# Configuration de l'API
 API_URL = "https://api-inference.huggingface.co/models/gpt2"
 api_token = os.environ.get("TOKEN")
 headers = {"Authorization": f"Bearer {api_token}"}
@@ -13,38 +11,42 @@ def query(payload):
     return response.json()
 
 def generate_yes_no_response(question):
-    # Créez un prompt qui demande explicitement une réponse Oui/Non
-    prompt = f"Réponds par 'Oui' ou 'Non' à la question suivante : {question}"
+    # Créez un prompt qui force le modèle à choisir entre Oui et Non
+    prompt = f"Question : {question}\nRéponse (Oui ou Non) :"
     
-    # Faites la requête à l'API
-    response = query({"inputs": prompt})
+    # Configurez les paramètres pour forcer une réponse courte
+    payload = {
+        "inputs": prompt,
+        "parameters": {
+            "max_new_tokens": 1,
+            "return_full_text": False,
+            "do_sample": True,
+            "top_k": 2,
+            "temperature": 0.7
+        }
+    }
     
-    # Extrayez le texte de la réponse
+    response = query(payload)
+    
+    # Assurez-vous que la réponse est soit "Oui" soit "Non"
     if isinstance(response, list) and len(response) > 0:
-        full_response = response[0].get('generated_text', '')
-    else:
-        full_response = str(response)  # Fallback si la réponse n'est pas comme prévu
+        answer = response[0].get('generated_text', '').strip().lower()
+        if answer == "oui":
+            return "Oui"
+        elif answer == "non":
+            return "Non"
     
-    # Analysez la réponse pour extraire Oui ou Non
-    if "oui" in full_response.lower():
-        return "Oui"
-    elif "non" in full_response.lower():
-        return "Non"
-    else:
-        # Si la réponse n'est ni oui ni non, choisissez aléatoirement
-        return random.choice(["Oui", "Non"])
+    # Si la réponse n'est pas "Oui" ou "Non", retournez une réponse par défaut
+    return "Réponse invalide"
 
-# Fonction pour l'interface Gradio
 def chatbot(message, history):
     response = generate_yes_no_response(message)
     return response
 
-# Création de l'interface Gradio
 iface = gr.ChatInterface(
     fn=chatbot,
     title="Chatbot Oui/Non",
     description="Posez une question, et je répondrai par Oui ou Non."
 )
 
-# Lancement de l'interface
 iface.launch()