Spaces:
Running
Running
admin
commited on
Commit
•
cf8b6d5
1
Parent(s):
6bc22ea
upd app
Browse files
app.py
CHANGED
@@ -9,7 +9,7 @@ from torchvision import transforms
|
|
9 |
from modelscope import snapshot_download
|
10 |
|
11 |
|
12 |
-
MODEL_DIR = snapshot_download("
|
13 |
|
14 |
|
15 |
def infer(input_img: str, checkpoint_file: str):
|
|
|
9 |
from modelscope import snapshot_download
|
10 |
|
11 |
|
12 |
+
MODEL_DIR = snapshot_download("MuGemSt/svhn", cache_dir="./__pycache__")
|
13 |
|
14 |
|
15 |
def infer(input_img: str, checkpoint_file: str):
|