acvss24 commited on
Commit
ab5914f
·
verified ·
1 Parent(s): 6159414

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -29,7 +29,7 @@ D = torch.device
29
  CPU = torch.device('cpu')
30
 
31
 
32
- current_directory = os.getcwd()
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')