low-res

app.py

@@ -74,7 +74,8 @@ def predict(img):
     upscale_factor = 4
     # lr_transforms = transforms.Resize((image_size[0]//upscale_factor, image_size[1]//upscale_factor), interpolation=IMode.BICUBIC, antialias=True)
     # to retain aspect ratio
-    lr_transforms = transforms.Resize(image_size[0]//upscale_factor, interpolation=IMode.BICUBIC, antialias=True)
+    lr_transforms = transforms.Resize(64, interpolation=IMode.BICUBIC, antialias=True)
+    #lr_transforms = transforms.Resize(image_size[0]//upscale_factor, interpolation=IMode.BICUBIC, antialias=True)
     # lr_transforms = transforms.Resize((128, 128), interpolation=IMode.BICUBIC, antialias=True)
     
     img = Image.fromarray(np.array(img))
@@ -149,4 +150,4 @@ gr.Interface(
         ["./demo_examples/head.png"],
         ["./demo_examples/woman.png"],
     ]
-).launch()
+).launch()