alex-abb commited on
Commit
563226d
·
verified ·
1 Parent(s): e0f3db5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -6
app.py CHANGED
@@ -2,8 +2,7 @@ import gradio as gr
2
  import spaces
3
  import time
4
  import os
5
- from transformers import AutoTokenizer, AutoModelForCausalLM
6
- import pytorch
7
  key = (os.getenv('API_KEY'))
8
 
9
  # Utiliser le jeton d'accès API pour charger le modèle
@@ -29,8 +28,5 @@ def generate_text(prompt):
29
 
30
  def chatbot(message, history):
31
  return generate_text(message)
32
- for i in range(len(message)):
33
- time.sleep(0.3)
34
- yield "You typed: " + message[: i+1]
35
-
36
  gr.ChatInterface(chatbot).launch()
 
2
  import spaces
3
  import time
4
  import os
5
+ from transformers import AutoTokenizer, AutoModelForCausalLM
 
6
  key = (os.getenv('API_KEY'))
7
 
8
  # Utiliser le jeton d'accès API pour charger le modèle
 
28
 
29
  def chatbot(message, history):
30
  return generate_text(message)
31
+
 
 
 
32
  gr.ChatInterface(chatbot).launch()