Spaces:

sneedium
/

dvatch_captcha_sneedium

Runtime error

sneedium commited on Nov 25, 2022

Commit

cfb287e

1 Parent(s): 91df712

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -11,7 +11,7 @@ from strhub.data.module import SceneTextDataModule
 # parseq = torch.load('tensor.pt', map_location=torch.device('cpu')).eval()
 from strhub.models.parseq.system import PARSeq as ModelClass
-parseq = ModelClass.load_from_checkpoint("./last.ckpt").eval()
 img_transform = SceneTextDataModule.get_transform(parseq.hparams.img_size)

 # parseq = torch.load('tensor.pt', map_location=torch.device('cpu')).eval()
 from strhub.models.parseq.system import PARSeq as ModelClass
+parseq = ModelClass.load_from_checkpoint("last.ckpt").eval()
 img_transform = SceneTextDataModule.get_transform(parseq.hparams.img_size)