Update modeling_prot2text.py
Browse files- modeling_prot2text.py +3 -3
modeling_prot2text.py
CHANGED
@@ -266,9 +266,9 @@ class Prot2TextModel(PreTrainedModel):
|
|
266 |
"dssp_config": DSSPConfig()}
|
267 |
config = ProteinGraphConfig(**config)
|
268 |
|
269 |
-
PATH_TO_DATA = f"
|
270 |
-
OUTPUT_FOLDER = f"
|
271 |
-
save_dir = f"
|
272 |
isExist = os.path.exists(PATH_TO_DATA)
|
273 |
if not isExist:
|
274 |
os.makedirs(PATH_TO_DATA)
|
|
|
266 |
"dssp_config": DSSPConfig()}
|
267 |
config = ProteinGraphConfig(**config)
|
268 |
|
269 |
+
PATH_TO_DATA = f"~/.tmp/pdb/pdb"
|
270 |
+
OUTPUT_FOLDER = f"~/.tmp/pdb/raw"
|
271 |
+
save_dir = f"~/.tmp/pdb/"
|
272 |
isExist = os.path.exists(PATH_TO_DATA)
|
273 |
if not isExist:
|
274 |
os.makedirs(PATH_TO_DATA)
|