Update app.py

app.py

@@ -1,4 +1,5 @@
 import gradio as gr
+from torchvision.transforms import Compose, Resize, ToTensor, Normalize,
 import matplotlib.pyplot as plt 
 from PIL import Image
 import numpy as np
@@ -236,8 +237,16 @@ class GeneratorUNet(nn.Module, HugGANModelHubMixin):
         return self.final(u7)
 
 def load_image_infer(image_file):
+    # Configure dataloaders
+    transform = Compose(
+            [
+                Resize((args.image_size, args.image_size), Image.BICUBIC),
+                ToTensor(),
+                Normalize((0.5, 0.5, 0.5), (0.5, 0.5, 0.5)),
+            ]
+        )
   image_file = Image.fromarray(np.array(image_file)[:, ::-1, :], "RGB")
-  image_file = normalize_test(image_file)
+  image_file = transform(image_file)
 
   return image_file