Update app.py

Outputting an image

app.py

@@ -20,11 +20,12 @@ def generate(seed):
     for i in range(generated_images.shape[0]):
         plt.subplot(8, 8, i+1)
         plt.imshow(post_process(generated_images[i, :, :, :]), cmap="binary")
-        plt.axis('off')
-    return fig
+        plt.axis("off")
+        plt.savefig("generated.png")
+    return "generated.png"
 
 gr.Interface(fn=generate, 
             inputs=[gr.inputs.Slider(label='Seed', minimum=0, maximum=1000, default=42)], 
-            outputs=gr.outputs.Image(type = "plot"), 
+            outputs="image", 
             title="CryptoGAN", 
             description="These CryptoPunks do not exist. gGnerate your own CryptoPunks").launch()