RamAnanth1 commited on
Commit
28425ed
1 Parent(s): 5914cbe

Update model.py

Browse files
Files changed (1) hide show
  1. model.py +1 -1
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