HassanDataSci commited on
Commit
99effa6
·
verified ·
1 Parent(s): 848db2f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -4,7 +4,7 @@ from transformers import pipeline
4
  # Load the conversational model
5
  @st.cache_resource
6
  def load_model():
7
- return pipeline("conversational", model="facebook/blenderbot-400M-distill")
8
 
9
  # Initialize the model
10
  generator = load_model()
 
4
  # Load the conversational model
5
  @st.cache_resource
6
  def load_model():
7
+ return pipeline("text-generation", model="facebook/blenderbot-400M-distill")
8
 
9
  # Initialize the model
10
  generator = load_model()