OutrageousOtter commited on
Commit
33b1ad9
·
verified ·
1 Parent(s): ae111d6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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='gpt2')
13
 
14
  # Fonction pour générer une réponse du chatbot
15
  def chatbot_response(user_input):
@@ -23,7 +23,7 @@ st.title("Expert en nutrition des volailles de chair et de ponte")
23
  st.markdown("### Posez une question à notre expert !")
24
 
25
  # Entrée utilisateur
26
- user_input = st.text_input("Vous :")
27
 
28
  if user_input:
29
  response = 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')
13
 
14
  # Fonction pour générer une réponse du chatbot
15
  def chatbot_response(user_input):
 
23
  st.markdown("### Posez une question à notre expert !")
24
 
25
  # Entrée utilisateur
26
+ user_input = st.text_input("")
27
 
28
  if user_input:
29
  response = chatbot_response(user_input)