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)
|