Update app.py
Browse files
app.py
CHANGED
@@ -47,7 +47,7 @@ def change_model_name(name):
|
|
47 |
MODEL_BUF["config"] = AutoConfig.from_pretrained(name)
|
48 |
|
49 |
|
50 |
-
def generate(
|
51 |
|
52 |
if model_name != MODEL_NAME:
|
53 |
change_model_name(model_name)
|
|
|
47 |
MODEL_BUF["config"] = AutoConfig.from_pretrained(name)
|
48 |
|
49 |
|
50 |
+
def generate(model_name, text):
|
51 |
|
52 |
if model_name != MODEL_NAME:
|
53 |
change_model_name(model_name)
|