Update app.py

app.py

@@ -50,11 +50,8 @@ def make_scaffold_motifs(pdb_code, start_idx, end_idx, scaffold_length, model_ty
         checkpoint = OA_DM_38M()
         model, collater, tokenizer, scheme = checkpoint
         data_top_dir = '/home/user/.cache/huggingface/datasets/'
-        # print("Current Directory Path:", os.getcwd())
-        # print("Folders in Current Directory:", os.listdir())
-        # print("Folders in User Directory:", os.listdir("/home/user/"))
+        os.makedirs(data_top_dir, exist_ok=True)
         # print("Folders in User Cache Directory:", os.listdir("/home/user/.cache"))
-        print("Folders in User Directory:", [ f.path for f in os.scandir("/home/user/") if f.is_dir() ])
         start_idx = list(map(int, start_idx.strip('][').split(',')))
         end_idx = list(map(int, end_idx.strip('][').split(',')))
         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')