Update app.py
Browse files
app.py
CHANGED
|
@@ -60,7 +60,7 @@ MODEL_NAME = 'microsoft/DialoGPT-small'# 'EleutherAI/gpt-neo-2.7B'#'distilbert/d
|
|
| 60 |
|
| 61 |
|
| 62 |
|
| 63 |
-
def load_model():
|
| 64 |
"""Download and load the model and tokenizer."""
|
| 65 |
try:
|
| 66 |
print(f"Loading model {MODEL_NAME}...")
|
|
|
|
| 60 |
|
| 61 |
|
| 62 |
|
| 63 |
+
def load_model(model_name):
|
| 64 |
"""Download and load the model and tokenizer."""
|
| 65 |
try:
|
| 66 |
print(f"Loading model {MODEL_NAME}...")
|