Update app.py
Browse files
app.py
CHANGED
|
@@ -75,7 +75,7 @@ def initialize_gpu():
|
|
| 75 |
pass
|
| 76 |
|
| 77 |
|
| 78 |
-
def reset_model(model_name):
|
| 79 |
# extract model info
|
| 80 |
model_args = deepcopy(model_info[model_name])
|
| 81 |
model_path = model_args.pop('model_path')
|
|
|
|
| 75 |
pass
|
| 76 |
|
| 77 |
|
| 78 |
+
def reset_model(model_name, *extra_components):
|
| 79 |
# extract model info
|
| 80 |
model_args = deepcopy(model_info[model_name])
|
| 81 |
model_path = model_args.pop('model_path')
|