Spaces:

awinml
/

2-qa-earnings-sentencewise

Build error

awinml commited on Apr 17, 2023

Commit

eb32817

•

1 Parent(s): 5482130

Upload utils.py

Files changed (1) hide show

utils.py CHANGED Viewed

@@ -36,8 +36,8 @@ def get_spacy_model():
 @st.experimental_singleton
 def get_flan_alpaca_xl_model():
-    model = AutoModelForSeq2SeqLM("./models/flan-alpaca-xl")
-    tokenizer = AutoTokenizer("./models/flan-alpaca-xl")
     return model, tokenizer

 @st.experimental_singleton
 def get_flan_alpaca_xl_model():
+    model = AutoModelForSeq2SeqLM.from_pretrained("./models/flan-alpaca-xl")
+    tokenizer = AutoTokenizer.from_pretrained("./models/flan-alpaca-xl")
     return model, tokenizer