use cpu

app.py

@@ -41,7 +41,7 @@ def download_model(url="https://huggingface.co/MuGeminorum/SVHN-Recognition/reso
 def _infer(path_to_checkpoint_file, path_to_input_image):
     model = Model()
     model.restore(path_to_checkpoint_file)
-    model.cuda()
+    # model.cuda()
     outstr = ''
 
     with torch.no_grad():
@@ -55,7 +55,7 @@ def _infer(path_to_checkpoint_file, path_to_input_image):
         image = Image.open(path_to_input_image)
         image = image.convert('RGB')
         image = transform(image)
-        images = image.unsqueeze(dim=0).cuda()
+        images = image.unsqueeze(dim=0)  # .cuda()
 
         length_logits, digit1_logits, digit2_logits, digit3_logits, digit4_logits, digit5_logits = model.eval()(images)