Spaces:

CYF200127
/

RxnIM

Sleeping

CYF200127 commited on 2 days ago

Commit

89d50ec

verified ·

1 Parent(s): ab71a14

Update rxn/reaction/interface.py

Files changed (1) hide show

rxn/reaction/interface.py CHANGED Viewed

@@ -77,7 +77,7 @@ class Reaction:
         return model
     def get_molscribe(self):
-        ckpt_path = hf_hub_download("yujieq/MolScribe", "swin_base_char_aux_1m.pth")
         molscribe = MolScribe(ckpt_path, device=self.device)
         return molscribe

         return model
     def get_molscribe(self):
+        ckpt_path = hf_hub_download("yujieq/MolScribe", "swin_base_char_aux_1m680k.pth")
         molscribe = MolScribe(ckpt_path, device=self.device)
         return molscribe