Image handling

app.py

@@ -61,14 +61,9 @@ def generate_random_walk(iters):
     plt.title('2D Random Walk, iterations = ' + str(iters))
     plt.grid()
     
-    # Save the plot to an in-memory buffer and return it
-    buf = io.BytesIO()
-    plt.savefig(buf, format='png')
-    buf.seek(0)
-    
-    return buf.read()
-
-# Create a Gradio interface
+    fig.canvas.draw()
+    image_array = np.array(fig.canvas.renderer.buffer_rgba())
+    return image_array
 
 
 iface = gr.Interface(fn=generate_random_walk, inputs=gr.inputs.Number(label="How many random steps"), outputs="image", title="Random Walk Plot")