Spaces:
Running
Running
Fix
Browse files
model.py
CHANGED
@@ -65,6 +65,7 @@ class Model:
|
|
65 |
self.task_name = ''
|
66 |
|
67 |
self.model_dir = pathlib.Path(model_dir)
|
|
|
68 |
|
69 |
self.use_lightweight = use_lightweight
|
70 |
if use_lightweight:
|
|
|
65 |
self.task_name = ''
|
66 |
|
67 |
self.model_dir = pathlib.Path(model_dir)
|
68 |
+
self.model_dir.mkdir(exist_ok=True, parents=True)
|
69 |
|
70 |
self.use_lightweight = use_lightweight
|
71 |
if use_lightweight:
|