Ellbendls commited on
Commit
7ddbd5f
·
1 Parent(s): 326d035

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -16,13 +16,13 @@ from langchain.chains import RetrievalQA
16
  from constants import CHROMA_SETTINGS
17
  from streamlit_chat import message
18
 
19
- use_auth_token
20
  load_dotenv(find_dotenv())
21
 
22
 
23
  st.set_page_config(layout="wide")
24
 
25
- checkpoint = "LaMini-Flan-T5-783M"
26
  tokenizer = AutoTokenizer.from_pretrained(checkpoint)
27
  base_model = AutoModelForSeq2SeqLM.from_pretrained(
28
  checkpoint,
 
16
  from constants import CHROMA_SETTINGS
17
  from streamlit_chat import message
18
 
19
+
20
  load_dotenv(find_dotenv())
21
 
22
 
23
  st.set_page_config(layout="wide")
24
 
25
+ checkpoint = ("LaMini-Flan-T5-783M", use_auth_token=True)
26
  tokenizer = AutoTokenizer.from_pretrained(checkpoint)
27
  base_model = AutoModelForSeq2SeqLM.from_pretrained(
28
  checkpoint,