Update app.py
Browse files
app.py
CHANGED
@@ -13,7 +13,7 @@ import datetime
|
|
13 |
|
14 |
# Setup
|
15 |
device = torch.device("cpu")
|
16 |
-
save_dir = "saved_predictions"
|
17 |
os.makedirs(save_dir, exist_ok=True)
|
18 |
|
19 |
# Load model
|
|
|
13 |
|
14 |
# Setup
|
15 |
device = torch.device("cpu")
|
16 |
+
save_dir = "/home/user/app/saved_predictions"
|
17 |
os.makedirs(save_dir, exist_ok=True)
|
18 |
|
19 |
# Load model
|