Update app.py

app.py

@@ -107,11 +107,10 @@ with image_blocks as demo:
 
     with gr.Row():
         with gr.Column():
-            image = gr.ImageEditor(sources=["upload"], layers=False, transforms=[])
-            # prompt = gr.Textbox(label="Prompt", placeholder="Enter your prompt here...")
+            image = gr.Image(label="Upload an image")
             with gr.Row(elem_id="prompt-container", equal_height=True):
-                with gr.Column():
-                    btn = gr.Button("Restyle!", elem_id="run_button")
+            with gr.Column():
+                btn = gr.Button("Restyle!", elem_id="run_button")
         
         with gr.Column():
             image_out = gr.Image(label="Output", elem_id="output-img")