Update app.py

app.py

@@ -46,22 +46,26 @@ def inference(img_path, model_name):
         if model_name == "4xGRL":
             weight_path = "pretrained/4x_APISR_GRL_GAN_generator.pth"
             auto_download_if_needed(weight_path)
-            generator = load_grl(weight_path, scale=4, device='cpu')
+            generator = load_grl(weight_path, scale=4)
+            generator = generator.to(device='cpu')
             
         elif model_name == "4xRRDB":
             weight_path = "pretrained/4x_APISR_RRDB_GAN_generator.pth"
             auto_download_if_needed(weight_path)
-            generator = load_rrdb(weight_path, scale=4, device='cpu')
+            generator = load_rrdb(weight_path, scale=4)
+            generator = generator.to(device='cpu')
             
         elif model_name == "2xRRDB":
             weight_path = "pretrained/2x_APISR_RRDB_GAN_generator.pth"
             auto_download_if_needed(weight_path)
-            generator = load_rrdb(weight_path, scale=2, device='cpu')
+            generator = load_rrdb(weight_path, scale=2)
+            generator = generator.to(device='cpu')
             
         elif model_name == "4xDAT":
             weight_path = "pretrained/4x_APISR_DAT_GAN_generator.pth"
             auto_download_if_needed(weight_path)
-            generator = load_dat(weight_path, scale=4, device='cpu')
+            generator = load_dat(weight_path, scale=4)
+            generator = generator.to(device='cpu')
             
         else:
             raise gr.Error("We don't support such Model")