Update app.py

app.py

@@ -106,10 +106,8 @@ class ImageTransform:
            ])}
    def __call__(self, img, phase='train'):
        img = self.transform[phase](img)
-​
        return img
-        
-   
+
 path = hf_hub_download('huggan/NeonGAN', 'model.bin')
 model_gen_n = torch.load(path, map_location=torch.device('cpu'))