Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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 |
-
|
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()
|