Update app.py

app.py

@@ -1,6 +1,7 @@
 import matplotlib.pyplot as plt
 import random
 import gradio as gr
+import io
 
 
 def generate_random_walk(iters):
@@ -66,5 +67,7 @@ def generate_random_walk(iters):
     return buf.read()
 
 # Create a Gradio interface
-iface = gr.Interface(fn=generate_random_walk, inputs="text", outputs="image", title="Random Walk Plot")
+
+
+iface = gr.Interface(fn=generate_random_walk, inputs=gr.inputs.Number(label="How many random steps"), outputs="image", title="Random Walk Plot")
 iface.launch()