update model

model.py

@@ -12,7 +12,7 @@ from time import process_time
 
 device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
 
-model_recog = load_from_checkpoint("weights/parseq/last.ckpt").eval().to(device)
+model_recog = load_from_checkpoint("weights/parseq/last.ckpt").eval().to(device).float()
 img_transform = SceneTextDataModule.get_transform(model_recog.hparams.img_size)
 
 model = torch.load('weights/best_model.pth', map_location=torch.device(device))
@@ -42,7 +42,7 @@ def prediction(image_path):
     
     image = cv2.cvtColor(result, cv2.COLOR_BGR2RGB)
     im_pil = Image.fromarray(image)
-    image = img_transform(im_pil).unsqueeze(0).to(device)
+    image = img_transform(im_pil).float().unsqueeze(0).to(device)
     
     p = model_recog(image).softmax(-1)
     pred, p = model_recog.tokenizer.decode(p)