Commit
•
f2540ce
1
Parent(s):
9ee88e4
Update app.py
Browse files
app.py
CHANGED
|
@@ -215,8 +215,7 @@ def main():
|
|
| 215 |
|
| 216 |
args = parse_args()
|
| 217 |
|
| 218 |
-
|
| 219 |
-
loaded_models["SwinV2"] = swin_model
|
| 220 |
|
| 221 |
tag_names, rating_indexes, general_indexes, character_indexes = load_labels()
|
| 222 |
|
|
|
|
| 215 |
|
| 216 |
args = parse_args()
|
| 217 |
|
| 218 |
+
change_model("SwinV2")
|
|
|
|
| 219 |
|
| 220 |
tag_names, rating_indexes, general_indexes, character_indexes = load_labels()
|
| 221 |
|