alex-abb commited on
Commit
a8cd301
·
verified ·
1 Parent(s): 3db2cf3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -1,8 +1,9 @@
1
  from transformers import pipeline, Conversation, ConversationalPipeline
2
  import gradio as gr
3
 
 
4
  # Charger le modèle GPT de Hugging Face
5
- model_id = "microsoft/DialoGPT-medium" # Remplacez par le modèle souhaité
6
  chat_pipeline = pipeline("conversational", model=model_id)
7
 
8
  # Consigne que le modèle suivra pour chaque chat
 
1
  from transformers import pipeline, Conversation, ConversationalPipeline
2
  import gradio as gr
3
 
4
+
5
  # Charger le modèle GPT de Hugging Face
6
+ model_id = "deepseek-ai/deepseek-llm-7b-chat" # Remplacez par le modèle souhaité
7
  chat_pipeline = pipeline("conversational", model=model_id)
8
 
9
  # Consigne que le modèle suivra pour chaque chat