Update app.py

app.py

@@ -3,7 +3,7 @@ import numpy as np
 import time
 
 # define core fn, which returns a generator {steps} times before returning the image
-def fake_diffusion(steps):
+def fake_diffusion(text, steps):
     for _ in range(steps):
         time.sleep(1)
         image = np.random.random((600, 600, 3))
@@ -13,7 +13,7 @@ def fake_diffusion(steps):
     yield image
 
 demo = gr.Interface(fake_diffusion, 
-                    inputs=gr.Slider(1, 10, 3), 
+                    inputs=[gr.Textbox(), gr.Slider(1, 10, 3)], 
                     outputs="image")
 
 # define queue - required for generators