alex-abb commited on
Commit
8172b02
·
verified ·
1 Parent(s): 48ba07e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +9 -0
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