update checkpoint

app.py

@@ -12,13 +12,13 @@ from indigo.renderer import IndigoRenderer
 from huggingface_hub import hf_hub_download
 
 REPO_ID = "yujieq/MolScribe"
-FILENAME = "swin_base_char_aux_200k.pth"
+FILENAME = "swin_base_char_aux_1m.pth"
 ckpt_path = hf_hub_download(REPO_ID, FILENAME)
 
 device = torch.device('cpu')
 model = MolScribe(ckpt_path, device)
 
-def generate_image(molblock):
+def generate_mol_image(molblock):
     indigo = Indigo()
     render = IndigoRenderer(indigo)
     indigo.setOption('render-output-format', 'png')
@@ -32,7 +32,7 @@ def generate_image(molblock):
 
 def predict(image):
     smiles, molblock = model.predict_image(image)
-    image = generate_image(molblock)
+    image = generate_mol_image(molblock)
     return image, smiles, molblock
 
 iface = gr.Interface(