Simon Duerr commited on
Commit
b405c72
1 Parent(s): d31b149
Files changed (1) hide show
  1. app.py +3 -0
app.py CHANGED
@@ -333,6 +333,9 @@ def protpardelle(path_to_file, m, resample_idx, modeltype, minlen, maxlen, step
333
  model.to(device)
334
  model.eval()
335
  model.device = device
 
 
 
336
 
337
  with open(save_dir + "/run_parameters.txt", "w") as f:
338
  f.write(f"Sampling run for {date_string}\n")
 
333
  model.to(device)
334
  model.eval()
335
  model.device = device
336
+
337
+ if config.train.home_dir == '':
338
+ config.train.home_dir = os.getcwd()
339
 
340
  with open(save_dir + "/run_parameters.txt", "w") as f:
341
  f.write(f"Sampling run for {date_string}\n")