Spaces:

teapotai
/

teapotchat

Running

zakerytclarke commited on Mar 23

Commit

88bcf19

verified ·

1 Parent(s): 0c5711c

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -20,6 +20,9 @@ import os
 from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
 with st.spinner('Loading Model...'):
     tokenizer = AutoTokenizer.from_pretrained(model_name)
     model = AutoModelForSeq2SeqLM.from_pretrained(model_name)

 from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
+tokenizer = None
+model = None
+model_name = "teapotai/teapotllm"
 with st.spinner('Loading Model...'):
     tokenizer = AutoTokenizer.from_pretrained(model_name)
     model = AutoModelForSeq2SeqLM.from_pretrained(model_name)