Update app.py

app.py

@@ -35,7 +35,7 @@ def generate_images (initial_archi, initial_class, batch_size):
     save_as_images(output)
     return output
     
- def convert_to_images(obj):
+def convert_to_images(obj):
     """ Convert an output tensor from BigGAN in a list of images.
         Params:
             obj: tensor or numpy array of shape (batch_size, channels, height, width)
@@ -59,7 +59,7 @@ def generate_images (initial_archi, initial_class, batch_size):
         img.append(PIL.Image.fromarray(out_array))
     return img
     
- def inference(initial_archi, initial_class):
+def inference(initial_archi, initial_class):
     output = generate_images (initial_archi, initial_class, 1)
     PIL_output = convert_to_images(output)
     return PIL_output[0]