alex-abb commited on
Commit
2989cb8
·
verified ·
1 Parent(s): 0464e17

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -3
app.py CHANGED
@@ -1,6 +1,6 @@
1
  import gradio as gr
2
  import spaces
3
- ## Load model directly
4
  import os
5
  from transformers import AutoTokenizer, AutoModelForCausalLM
6
 
@@ -28,8 +28,11 @@ def generate_text(prompt):
28
 
29
  return response_text
30
  # Définir une fonction pour l'interface de chat
31
- def chatbot(message,history):
32
- return generate_text(message)
33
 
 
 
 
 
 
34
 
35
  gr.ChatInterface(chatbot).launch()
 
1
  import gradio as gr
2
  import spaces
3
+ import time
4
  import os
5
  from transformers import AutoTokenizer, AutoModelForCausalLM
6
 
 
28
 
29
  return response_text
30
  # Définir une fonction pour l'interface de chat
 
 
31
 
32
+ def chatbot(message, history):
33
+ return generate_text(message)
34
+ for i in range(len(message)):
35
+ time.sleep(0.3)
36
+ yield "You typed: " + message[: i+1]
37
 
38
  gr.ChatInterface(chatbot).launch()