sylvielsstfr commited on
Commit
bb85704
·
1 Parent(s): 2b79c44
Files changed (1) hide show
  1. app.py +9 -13
app.py CHANGED
@@ -1,28 +1,24 @@
1
  import gradio as gr
2
  from transformers import pipeline
3
 
4
- # Optionnel : modèle public GPT-2 pour des réponses réelles
5
  generator = pipeline("text-generation", model="gpt2")
6
 
7
  def respond(message, history):
8
  history = history or []
9
-
10
- # Ajouter message utilisateur
11
- user_msg = message
12
- # Générer réponse via GPT-2
13
  answer = generator(message, max_length=50, do_sample=True)[0]["generated_text"]
14
-
15
- # Ajouter à l'historique sous forme tuple (ancien template 3.x)
16
- history.append((user_msg, answer))
17
-
 
18
  return history
19
 
20
- # Interface Gradio 3.43
21
  iface = gr.ChatInterface(respond, title="Chat Demo")
22
  iface.launch()
23
 
24
 
25
 
26
-
27
-
28
-
 
1
  import gradio as gr
2
  from transformers import pipeline
3
 
4
+ # GPT-2 public
5
  generator = pipeline("text-generation", model="gpt2")
6
 
7
  def respond(message, history):
8
  history = history or []
9
+
10
+ # Générer la réponse
 
 
11
  answer = generator(message, max_length=50, do_sample=True)[0]["generated_text"]
12
+
13
+ # Ajouter le tuple (user, bot) à l'historique
14
+ history.append((message, answer))
15
+
16
+ # Retourner la liste de tuples
17
  return history
18
 
19
+ # Gradio 3.43
20
  iface = gr.ChatInterface(respond, title="Chat Demo")
21
  iface.launch()
22
 
23
 
24