Ellbendls commited on
Commit
6390b65
1 Parent(s): 95d8345

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -22,7 +22,7 @@ load_dotenv(find_dotenv())
22
 
23
  st.set_page_config(layout="wide")
24
 
25
- checkpoint = "OpenAssistant/oasst-sft-1-pythia-12b"
26
  tokenizer = AutoTokenizer.from_pretrained(checkpoint)
27
  base_model = AutoModelForSeq2SeqLM.from_pretrained(
28
  checkpoint,
 
22
 
23
  st.set_page_config(layout="wide")
24
 
25
+ checkpoint = "google/flan-ul2"
26
  tokenizer = AutoTokenizer.from_pretrained(checkpoint)
27
  base_model = AutoModelForSeq2SeqLM.from_pretrained(
28
  checkpoint,