Hugging Face's logo

Spaces:
shaneperry0101
/
streamlit-chat-app
Sleeping

App Files Community
shaneperry0101 commited on
Commit
1fe3c38
·
verified ·
1 Parent(s): 10a5cde

fix(model): test switch the model

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -3,7 +3,8 @@ from transformers import pipeline
3
 
4
 
5
  # Load the conversational model from Hugging Face
6
- chatbot = pipeline('text-generation', model='shaneperry0101/Health-Llama-3.2-1B')
 
7
 
8
  # Initialize session state for storing the conversation
9
  if 'conversation' not in st.session_state:
 
3
 
4
 
5
  # Load the conversational model from Hugging Face
6
+ # chatbot = pipeline('text-generation', model='shaneperry0101/Health-Llama-3.2-1B')
7
+ chatbot = pipeline('text-generation', model='microsoft/DialoGPT-medium')
8
 
9
  # Initialize session state for storing the conversation
10
  if 'conversation' not in st.session_state: