Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -1,6 +1,12 @@
|
|
1 |
from transformers import pipeline, Conversation
|
2 |
import gradio as gr
|
3 |
import os
|
|
|
|
|
|
|
|
|
|
|
|
|
4 |
|
5 |
# Charger le modèle GPT de Hugging Face
|
6 |
model_id = "deepseek-ai/deepseek-llm-7b-chat"
|
@@ -9,6 +15,9 @@ pipe = pipeline("text-generation", model=model_id)
|
|
9 |
# Consigne que le modèle suivra pour chaque chat
|
10 |
system_message = "You are a helpful assistant that always responds in a friendly manner."
|
11 |
|
|
|
|
|
|
|
12 |
# Fonction pour générer une réponse à partir du message de l'utilisateur
|
13 |
def generate_response(user_message, history):
|
14 |
# Ajouter le message système au début de l'historique des messages
|
|
|
1 |
from transformers import pipeline, Conversation
|
2 |
import gradio as gr
|
3 |
import os
|
4 |
+
import spaces
|
5 |
+
|
6 |
+
|
7 |
+
|
8 |
+
|
9 |
+
|
10 |
|
11 |
# Charger le modèle GPT de Hugging Face
|
12 |
model_id = "deepseek-ai/deepseek-llm-7b-chat"
|
|
|
15 |
# Consigne que le modèle suivra pour chaque chat
|
16 |
system_message = "You are a helpful assistant that always responds in a friendly manner."
|
17 |
|
18 |
+
|
19 |
+
@spaces.GPU
|
20 |
+
|
21 |
# Fonction pour générer une réponse à partir du message de l'utilisateur
|
22 |
def generate_response(user_message, history):
|
23 |
# Ajouter le message système au début de l'historique des messages
|