alex-abb commited on
Commit
93116dd
1 Parent(s): 13559a3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -6
app.py CHANGED
@@ -12,9 +12,9 @@ def query(payload):
12
 
13
  def generate_yes_no_response(question):
14
  # Créez un prompt qui force le modèle à choisir entre Oui et Non
15
- prompt = f"Question : {question}\nRéponse (Oui ou Non) :"
16
 
17
- # Configurez les paramètres pour forcer une réponse courte
18
  payload = {
19
  "inputs": prompt,
20
  "parameters": {
@@ -31,10 +31,10 @@ def generate_yes_no_response(question):
31
  # Assurez-vous que la réponse est soit "Oui" soit "Non"
32
  if isinstance(response, list) and len(response) > 0:
33
  answer = response[0].get('generated_text', '').strip().lower()
34
- if answer == "oui":
35
- return "Oui"
36
- elif answer == "non":
37
- return "Non"
38
 
39
  # Si la réponse n'est pas "Oui" ou "Non", retournez une réponse par défaut
40
  return "Réponse invalide"
 
12
 
13
  def generate_yes_no_response(question):
14
  # Créez un prompt qui force le modèle à choisir entre Oui et Non
15
+ prompt = f" {question}\nanswer only yes or no :"
16
 
17
+ # Configurez les paramètres pour forcer une réponse courte
18
  payload = {
19
  "inputs": prompt,
20
  "parameters": {
 
31
  # Assurez-vous que la réponse est soit "Oui" soit "Non"
32
  if isinstance(response, list) and len(response) > 0:
33
  answer = response[0].get('generated_text', '').strip().lower()
34
+ if answer == "yes":
35
+ return "yes"
36
+ elif answer == "no":
37
+ return "no"
38
 
39
  # Si la réponse n'est pas "Oui" ou "Non", retournez une réponse par défaut
40
  return "Réponse invalide"