Hugging Face's logo

Spaces:
gaia-mistral
/
volAI_Avril
Running

App Files Community
OutrageousOtter commited on
Commit
fe4b91d
·
verified ·
1 Parent(s): 33b1ad9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -9,7 +9,7 @@ import tensorflow
9
  from transformers import pipeline
10
 
11
  # Chargement du modèle Hugging Face (GPT-2 ou autre modèle de génération de texte)
12
- generator = pipeline('text-generation', model='HuggingFaceH4/zephyr-7b-beta')
13
 
14
  # Fonction pour générer une réponse du chatbot
15
  def chatbot_response(user_input):
 
9
  from transformers import pipeline
10
 
11
  # Chargement du modèle Hugging Face (GPT-2 ou autre modèle de génération de texte)
12
+ generator = pipeline("text-generation", model="HuggingFaceH4/zephyr-7b-beta", torch_dtype=torch.bfloat16, device_map="auto")
13
 
14
  # Fonction pour générer une réponse du chatbot
15
  def chatbot_response(user_input):