init

app.py

@@ -52,7 +52,7 @@ def load_checkpoint(model, ckpt_path):
         new_state_dict[k] = v
 
     model.load_state_dict(new_state_dict)
-    # model.cuda()
+    model.cuda()
     del checkpoint
     torch.cuda.empty_cache()
 
@@ -77,8 +77,8 @@ def image_infer(img_PIL):
         trainer.model.eval()
         img_lr, img_lr_up = get_img_data(img_PIL, hparams, sr_scale=4)
 
-        # img_lr = img_lr.to('cuda')
-        # img_lr_up = img_lr_up.to('cuda')
+        img_lr = img_lr.to('cuda')
+        img_lr_up = img_lr_up.to('cuda')
 
         img_sr, _ = trainer.model.sample(img_lr, img_lr_up, img_lr_up.shape)