feat: Refactor app.py to remove unused code, improve code structure, and add prediction functionality

app.py

@@ -27,25 +27,27 @@ with gr.Blocks() as demo:
       input_image = gr.Image(type="pil", label="Input Image")
       run_button = gr.Button("Enhance Image")
     with gr.Column():
-      output_slider = ImageSlider(label="Before / After", type="numpy")
-    with gr.Accordion("Advanced Options", open=False):
-      resolution = gr.Slider(minimum=128, maximum=2048, value=1024, step=128, label="Resolution")
-      num_inference_steps = gr.Slider(minimum=1, maximum=50, value=20, step=1, label="Number of Inference Steps")
-      strength = gr.Slider(minimum=0, maximum=1, value=0.2, step=0.01, label="Strength")
-      hdr = gr.Slider(minimum=0, maximum=1, value=0, step=0.1, label="HDR Effect")
-      guidance_scale = gr.Slider(minimum=0, maximum=20, value=6, step=0.5, label="Guidance Scale")
-      controlnet_strength = gr.Slider(minimum=0.0, maximum=2.0, value=0.75, step=0.05, label="ControlNet Strength")
-      scheduler_name = gr.Dropdown(
-          choices=["DDIM", "DPM++ 3M SDE Karras", "DPM++ 3M Karras"],
-          value="DDIM",
-          label="Scheduler"
-      )
-
-
-    # run_button.click(
-    #   fn=gradio_process_image, 
-    #   inputs=[input_image, resolution, num_inference_steps, strength, hdr, guidance_scale, controlnet_strength, scheduler_name],
-    #   outputs=output_slider
-    # )
+      output_image = gr.Image(type="pil", label="Output Image")
+      # output_slider = ImageSlider(label="Before / After", type="numpy")
+
+  with gr.Accordion("Advanced Options", open=False):
+    resolution = gr.Slider(minimum=128, maximum=2048, value=1024, step=128, label="Resolution")
+    num_inference_steps = gr.Slider(minimum=1, maximum=50, value=20, step=1, label="Number of Inference Steps")
+    strength = gr.Slider(minimum=0, maximum=1, value=0.2, step=0.01, label="Strength")
+    hdr = gr.Slider(minimum=0, maximum=1, value=0, step=0.1, label="HDR Effect")
+    guidance_scale = gr.Slider(minimum=0, maximum=20, value=6, step=0.5, label="Guidance Scale")
+    controlnet_strength = gr.Slider(minimum=0.0, maximum=2.0, value=0.75, step=0.05, label="ControlNet Strength")
+    scheduler_name = gr.Dropdown(
+        choices=["DDIM", "DPM++ 3M SDE Karras", "DPM++ 3M Karras"],
+        value="DDIM",
+        label="Scheduler"
+    )
+
+
+    run_button.click(
+      fn=gradio_process_image, 
+      inputs=[input_image, resolution, num_inference_steps, strength, hdr, guidance_scale, controlnet_strength, scheduler_name],
+      outputs=output_image
+    )
     
 demo.launch()