Update app.py

app.py

@@ -238,17 +238,15 @@ class GeneratorUNet(nn.Module, HugGANModelHubMixin):
 
 def load_image_infer(image_file):
     # Configure dataloaders
-    transform = Compose(
-            [
+    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 = transform(image_file)
+            ])
+    image_file = Image.fromarray(np.array(image_file)[:, ::-1, :], "RGB")
+    image_file = transform(image_file)
 
-  return image_file
+    return image_file
   
 def generate_images(test_input):
   test_input = load_image_infer(test_input)