Spaces:
Runtime error
Runtime error
RamAnanth1
commited on
Commit
•
28425ed
1
Parent(s):
5914cbe
Update model.py
Browse files
model.py
CHANGED
@@ -53,7 +53,7 @@ class Model:
|
|
53 |
use_lightweight: bool = True):
|
54 |
self.device = torch.device(
|
55 |
'cuda:0' if torch.cuda.is_available() else 'cpu')
|
56 |
-
self.model_dir = model_dir
|
57 |
|
58 |
self.download_models()
|
59 |
|
|
|
53 |
use_lightweight: bool = True):
|
54 |
self.device = torch.device(
|
55 |
'cuda:0' if torch.cuda.is_available() else 'cpu')
|
56 |
+
self.model_dir = pathlib.Path(model_dir)
|
57 |
|
58 |
self.download_models()
|
59 |
|