Update app.py

app.py

@@ -19,11 +19,11 @@ def generate(seed):
         plt.imshow(generated_images[i, :, :, :])
         plt.axis("off")
     plt.savefig('foo.png')
-    return fig
+    return "foo.png"
 
 
 gr.Interface(fn=generate, 
             inputs=[gr.inputs.Slider(label='Seed', minimum=0, maximum=1000, default=42)], 
-            outputs=gr.Plot(label="Plot"), 
+            outputs=gr.Image(), 
             title="CryptoGAN", 
             description="These CryptoPunks do not exist.").launch()