Spaces:
Running
Running
Update app.py
Browse files
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 = {
|