init

app.py

@@ -107,7 +107,7 @@ def tensor2im(var):
      return var.mul(t_stds).add(t_means).mul(255.).clamp(0,255).permute(1,2,0)
 
 def proc_pil_img(input_image, model):
-    transformed_image = img_transforms(input_image)[None,...].cuda().half()
+    transformed_image = img_transforms(input_image)[None,...].cpu().half().float()
             
     with torch.no_grad():
         result_image = model(transformed_image)[0]