sabssag commited on
Commit
e75a48b
·
verified ·
1 Parent(s): 416be73

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -2,7 +2,7 @@ import streamlit as st
2
  from transformers import pipeline
3
 
4
  # Load the conversational pipeline
5
- model_name = "QAModel"
6
  chatbot_pipeline = pipeline("conversational", model=model_name, tokenizer=model_name)
7
 
8
  # Set the title for the Streamlit app
 
2
  from transformers import pipeline
3
 
4
  # Load the conversational pipeline
5
+ model_name = "./QAModel"
6
  chatbot_pipeline = pipeline("conversational", model=model_name, tokenizer=model_name)
7
 
8
  # Set the title for the Streamlit app