Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -29,7 +29,7 @@ D = torch.device
|
|
| 29 |
CPU = torch.device('cpu')
|
| 30 |
|
| 31 |
|
| 32 |
-
current_directory =
|
| 33 |
save_path = os.path.join(os.path.dirname(current_directory), "pretrained_models")
|
| 34 |
os.makedirs(save_path, exist_ok=True)
|
| 35 |
model_path = os.path.join(save_path, 'model_weights.pt')
|
|
|
|
| 29 |
CPU = torch.device('cpu')
|
| 30 |
|
| 31 |
|
| 32 |
+
current_directory = "/home/user/app/"
|
| 33 |
save_path = os.path.join(os.path.dirname(current_directory), "pretrained_models")
|
| 34 |
os.makedirs(save_path, exist_ok=True)
|
| 35 |
model_path = os.path.join(save_path, 'model_weights.pt')
|