Update app.py

app.py

@@ -38,8 +38,9 @@ with gr.Blocks() as demo:
     
     with gr.Row():
         with gr.Column():
-            canvas = gr.Sketchpad(label="Draw Here")
+            canvas = gr.Sketchpad(label="Draw Here", tool="brush")
             clear_btn = gr.Button("Clear")
+            generate_btn = gr.Button("Generate Image")
         
         with gr.Column():
             # Add color buttons for terrains
@@ -56,7 +57,7 @@ with gr.Blocks() as demo:
         color_btns[color_name].click(fn=lambda color=color: canvas.update(value=color), inputs=None, outputs=None)
 
     # Click to generate an image
-    canvas.submit(fn=predict, inputs=canvas, outputs=output_image)
+    generate_btn.click(fn=predict, inputs=canvas, outputs=output_image)
 
 # Launch the Gradio app
 demo.launch()