colbyford commited on
Commit
6e96edf
·
1 Parent(s): 76f2fb3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -49,8 +49,9 @@ def make_scaffold_motifs(pdb_code, start_idx, end_idx, scaffold_length, model_ty
49
  if model_type == "EvoDiff-Seq":
50
  checkpoint = OA_DM_38M()
51
  model, collater, tokenizer, scheme = checkpoint
52
- data_top_dir = ''
53
  print("Current Scaffold Directory:", os.getcwd())
 
54
  start_idx = list(map(int, start_idx.strip('][').split(',')))
55
  end_idx = list(map(int, end_idx.strip('][').split(',')))
56
  generated_sequence, new_start_idx, new_end_idx = generate_scaffold(model, pdb_code, start_idx, end_idx, scaffold_length, data_top_dir, tokenizer, device='cpu')
 
49
  if model_type == "EvoDiff-Seq":
50
  checkpoint = OA_DM_38M()
51
  model, collater, tokenizer, scheme = checkpoint
52
+ data_top_dir = '/home/user/app/'
53
  print("Current Scaffold Directory:", os.getcwd())
54
+ print("Folders in Current Directory:", os.listdir())
55
  start_idx = list(map(int, start_idx.strip('][').split(',')))
56
  end_idx = list(map(int, end_idx.strip('][').split(',')))
57
  generated_sequence, new_start_idx, new_end_idx = generate_scaffold(model, pdb_code, start_idx, end_idx, scaffold_length, data_top_dir, tokenizer, device='cpu')