alex-abb commited on
Commit
3e84f43
·
verified ·
1 Parent(s): d37ef41

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -2
app.py CHANGED
@@ -6,9 +6,13 @@ import spaces
6
 
7
  @space.gpu
8
  # Charger le modèle GPT de Hugging Face
9
- model_id = "deepseek-ai/deepseek-llm-7b-chat" # Remplacez par le modèle souhaité
10
- chat_pipeline = pipeline("conversational", model=model_id)
11
 
 
 
 
 
 
12
  # Consigne que le modèle suivra pour chaque chat
13
  system_message = "You are a helpful assistant that always responds in a friendly manner."
14
 
 
6
 
7
  @space.gpu
8
  # Charger le modèle GPT de Hugging Face
9
+ # Use a pipeline as a high-level helper
 
10
 
11
+ messages = [
12
+ {"role": "user", "content": "Who are you?"},
13
+ ]
14
+ pipe = pipeline("text-generation", model="deepseek-ai/deepseek-llm-7b-chat")
15
+ pipe(messages)
16
  # Consigne que le modèle suivra pour chaque chat
17
  system_message = "You are a helpful assistant that always responds in a friendly manner."
18