varun500 commited on
Commit
770a02f
1 Parent(s): 6ac8498

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -11,9 +11,9 @@ from transformers import (
11
 
12
  from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
13
 
14
- tokenizer = AutoTokenizer.from_pretrained("declare-lab/flan-alpaca-large")
15
 
16
- model = AutoModelForSeq2SeqLM.from_pretrained("declare-lab/flan-alpaca-large")
17
  def main():
18
  # Set app title
19
  st.title("Flan Alpaca Large Model")
 
11
 
12
  from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
13
 
14
+ tokenizer = AutoTokenizer.from_pretrained("declare-lab/flan-alpaca-base")
15
 
16
+ model = AutoModelForSeq2SeqLM.from_pretrained("declare-lab/flan-alpaca-base")
17
  def main():
18
  # Set app title
19
  st.title("Flan Alpaca Large Model")