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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -0
app.py CHANGED
@@ -1,7 +1,10 @@
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)
 
1
  from transformers import pipeline, Conversation, ConversationalPipeline
2
  import gradio as gr
3
 
4
+ import spaces
5
 
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)