Ellbendls commited on
Commit
95d8345
·
1 Parent(s): e669e79

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 = "MBZUAI/LaMini-T5-738M"
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 = "OpenAssistant/oasst-sft-1-pythia-12b"
26
  tokenizer = AutoTokenizer.from_pretrained(checkpoint)
27
  base_model = AutoModelForSeq2SeqLM.from_pretrained(
28
  checkpoint,