Spaces:

yseop
/

Finance

Runtime error

dmariko commited on Oct 26, 2021

Commit

4c507d7

1 Parent(s): 2f7fcf5

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -43,7 +43,7 @@ def change_model_name(name):
     MODEL_BUF["name"] = name
     MODEL_BUF["tokenizer"] = AutoTokenizer.from_pretrained(name)
-    MODEL_BUF["model"] = AutoModelForSequenceClassification.from_pretrained(name)
     MODEL_BUF["config"] = AutoConfig.from_pretrained(name)

     MODEL_BUF["name"] = name
     MODEL_BUF["tokenizer"] = AutoTokenizer.from_pretrained(name)
+    MODEL_BUF["model"] = AutoModelForSeq2SeqLM.from_pretrained(name)
     MODEL_BUF["config"] = AutoConfig.from_pretrained(name)