Update app.py
Browse files
app.py
CHANGED
@@ -43,7 +43,7 @@ def change_model_name(name):
|
|
43 |
|
44 |
MODEL_BUF["name"] = name
|
45 |
MODEL_BUF["tokenizer"] = AutoTokenizer.from_pretrained(name)
|
46 |
-
MODEL_BUF["model"] =
|
47 |
MODEL_BUF["config"] = AutoConfig.from_pretrained(name)
|
48 |
|
49 |
|
|
|
43 |
|
44 |
MODEL_BUF["name"] = name
|
45 |
MODEL_BUF["tokenizer"] = AutoTokenizer.from_pretrained(name)
|
46 |
+
MODEL_BUF["model"] = AutoModelForSeq2SeqLM.from_pretrained(name)
|
47 |
MODEL_BUF["config"] = AutoConfig.from_pretrained(name)
|
48 |
|
49 |
|