Update app.py

app.py

@@ -8,10 +8,12 @@ n_images = 64
 codings_size = 100
 hub_generator = from_pretrained_keras("huggan/crypto-gan")
 
+
 def post_process(image):
     image = ((image + 1) / 2) # scale back
     return image
 
+
 def generate(seed):
     noise = tf.random.normal(shape=[n_images, codings_size], seed=seed)
     generated_images = hub_generator(noise, training=False)
@@ -21,11 +23,11 @@ def generate(seed):
         plt.subplot(8, 8, i+1)
         plt.imshow(post_process(generated_images[i, :, :, :]), cmap="binary")
         plt.axis("off")
-        plt.savefig("generated.png")
-    return "generated.png"
+    return fig
+
 
 gr.Interface(fn=generate, 
             inputs=[gr.inputs.Slider(label='Seed', minimum=0, maximum=1000, default=42)], 
-            outputs="image", 
+            outputs=gr.outputs.Image(type="plot"), 
             title="CryptoGAN", 
             description="These CryptoPunks do not exist. gGnerate your own CryptoPunks").launch()