Hellisotherpeople commited on
Commit
62e3c76
β€’
1 Parent(s): 27f1fea
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -65,7 +65,7 @@ form.form_submit_button("Generate some Constrained Text!")
65
 
66
  with st.spinner("Please wait while the model loads:"):
67
  tokenizer = AutoTokenizer.from_pretrained(model_name)
68
- model = AutoModelForCausalLM.from_pretrained(model_name)
69
 
70
 
71
  def isPalindrome(s):
 
65
 
66
  with st.spinner("Please wait while the model loads:"):
67
  tokenizer = AutoTokenizer.from_pretrained(model_name)
68
+ model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
69
 
70
 
71
  def isPalindrome(s):