update image display

app.py

@@ -1,6 +1,5 @@
 import gradio as gr
 import random
-from PIL import Image
 
 # Define the initial options
 options = ["pink"] * 15 + ["blue"] * 10
@@ -21,15 +20,15 @@ def draw():
         if options:
             option = spin_roulette()
             if option == "pink":
-                pink = Image.open("images/pink.png")
+                pink = "images/pink.png"
                 return pink
             else:
-                blue = Image.open("images/blue.png")
+                blue = "images/blue.png"
                 return blue
 
 
 with gr.Blocks() as demo:
     result = draw()
     draw_btn = gr.Button("Draw")
-    draw_btn.click(fn=draw, inputs=[], outputs=result)
+    draw_btn.click(fn=draw, inputs=[], outputs=gr.Image(value=result))
 demo.launch()