CYF200127 commited on
Commit
89d50ec
·
verified ·
1 Parent(s): ab71a14

Update rxn/reaction/interface.py

Browse files
Files changed (1) hide show
  1. rxn/reaction/interface.py +1 -1
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", "swin_base_char_aux_1m.pth")
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