Update rxn/reaction/interface.py
Browse files
rxn/reaction/interface.py
CHANGED
@@ -77,7 +77,7 @@ class Reaction:
|
|
77 |
return model
|
78 |
|
79 |
def get_molscribe(self):
|
80 |
-
ckpt_path = hf_hub_download("yujieq/MolScribe", "
|
81 |
molscribe = MolScribe(ckpt_path, device=self.device)
|
82 |
return molscribe
|
83 |
|
|
|
77 |
return model
|
78 |
|
79 |
def get_molscribe(self):
|
80 |
+
ckpt_path = hf_hub_download("yujieq/MolScribe", "swin_base_char_aux_1m680k.pth")
|
81 |
molscribe = MolScribe(ckpt_path, device=self.device)
|
82 |
return molscribe
|
83 |
|