Harry2406 commited on
Commit
6b79409
·
verified ·
1 Parent(s): b4140a5

Update src/streamlit_app.py

Browse files
Files changed (1) hide show
  1. src/streamlit_app.py +2 -2
src/streamlit_app.py CHANGED
@@ -15,8 +15,8 @@ st.set_page_config(
15
  # Load Model
16
  # =========================
17
  def load_model():
18
- tokenizer = AutoTokenizer.from_pretrained("google/flan-t5-Large")
19
- model = AutoModelForSeq2SeqLM.from_pretrained("google/flan-t5-Large")
20
  return tokenizer, model
21
 
22
  tokenizer, model = load_model()
 
15
  # Load Model
16
  # =========================
17
  def load_model():
18
+ tokenizer = AutoTokenizer.from_pretrained("google/flan-t5-large")
19
+ model = AutoModelForSeq2SeqLM.from_pretrained("google/flan-t5-large")
20
  return tokenizer, model
21
 
22
  tokenizer, model = load_model()