colbyford commited on
Commit
c7cc63a
1 Parent(s): a6c6c9c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -49,8 +49,8 @@ def make_scaffold_motifs(pdb_code, start_idx, end_idx, scaffold_length, model_ty
49
  checkpoint = OA_DM_38M()
50
  model, collater, tokenizer, scheme = checkpoint
51
  data_top_dir = ''
52
- start_idx = list(map(int, start_idx.strip('][').split(', ')))
53
- end_idx = list(map(int, end_idx.strip('][').split(', ')))
54
  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')
55
 
56
  generated_scaffold_output = {
 
49
  checkpoint = OA_DM_38M()
50
  model, collater, tokenizer, scheme = checkpoint
51
  data_top_dir = ''
52
+ start_idx = list(map(int, start_idx.strip('][').split(',')))
53
+ end_idx = list(map(int, end_idx.strip('][').split(',')))
54
  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')
55
 
56
  generated_scaffold_output = {