alex-abb commited on
Commit
72e0722
1 Parent(s): 93116dd

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -14
app.py CHANGED
@@ -10,9 +10,9 @@ def query(payload):
10
  response = requests.post(API_URL, headers=headers, json=payload)
11
  return response.json()
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 = {
@@ -28,19 +28,9 @@ def generate_yes_no_response(question):
28
 
29
  response = query(payload)
30
 
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"
41
-
42
  def chatbot(message, history):
43
- response = generate_yes_no_response(message)
44
  return response
45
 
46
  iface = gr.ChatInterface(
 
10
  response = requests.post(API_URL, headers=headers, json=payload)
11
  return response.json()
12
 
13
+ def reponse(question):
14
  # Créez un prompt qui force le modèle à choisir entre Oui et Non
15
+ prompt = {question}
16
 
17
  # Configurez les paramètres pour forcer une réponse courte
18
  payload = {
 
28
 
29
  response = query(payload)
30
 
31
+
 
 
 
 
 
 
 
 
 
 
32
  def chatbot(message, history):
33
+ response = reponse(message)
34
  return response
35
 
36
  iface = gr.ChatInterface(