Update app.py

app.py

@@ -22,7 +22,6 @@ import functools
 import os
 import tempfile
 
-import diffusers
 import gradio as gr
 import imageio as imageio
 import numpy as np
@@ -50,10 +49,7 @@ class Examples(gradio.helpers.Examples):
 default_seed = 2024
 default_batch_size = 1
 
-default_image_processing_resolution = 768
-
-default_video_num_inference_steps = 10
-default_video_processing_resolution = 768
+default_image_processing_resolution = 2048
 default_video_out_max_frames = 60
 
 def process_image_check(path_input):
@@ -99,12 +95,10 @@ def process_image(
     path_output_dir = tempfile.mkdtemp()
     path_out_png = os.path.join(path_output_dir, f"{name_base}_delight.png")
     input_image = Image.open(path_input)
-    input_image = resize_image(input_image, default_image_processing_resolution)
-
     pipe_out = pipe(
         input_image,
         match_input_resolution=False,
-        processing_resolution=max(input_image.size)
+        processing_resolution=default_image_processing_resolution
     )
 
     processed_frame = (pipe_out.prediction.clip(-1, 1) + 1) / 2
@@ -113,20 +107,6 @@ def process_image(
     processed_frame.save(path_out_png)
     yield [input_image, path_out_png]
 
-def center_crop(img):
-    # Open the image file
-    img_width, img_height = img.size
-    crop_width =min(img_width, img_height)
-    # Calculate the cropping box
-    left = (img_width - crop_width) / 2
-    top = (img_height - crop_width) / 2
-    right = (img_width + crop_width) / 2
-    bottom = (img_height + crop_width) / 2
-    
-    # Crop the image
-    img_cropped = img.crop((left, top, right, bottom))
-    return img_cropped
-
 def process_video(
     pipe,
     path_input,
@@ -143,7 +123,7 @@ def process_video(
     print(f"Processing video {name_base}{name_ext}")
 
     path_output_dir = tempfile.mkdtemp()
-    path_out_vis = os.path.join(path_output_dir, f"{name_base}_delight_colored.mp4")
+    path_out_vis = os.path.join(path_output_dir, f"{name_base}_delight.mp4")
 
     init_latents = None
     reader, writer = None, None
@@ -170,11 +150,11 @@ def process_video(
                 break
 
             frame_pil = Image.fromarray(frame)
-            # frame_pil = center_crop(frame_pil)
             pipe_out = pipe(
                 frame_pil,
                 match_input_resolution=False,
-                latents=init_latents
+                latents=init_latents,
+                processing_resolution=default_image_processing_resolution
             )
 
             if init_latents is None:
@@ -212,7 +192,7 @@ def run_demo_server(pipe):
 
     with gr.Blocks(
         theme=gradio_theme,
-        title="StableDelight",
+        title="Stable Delight Estimation",
         css="""
             #download {
                 height: 118px;
@@ -256,7 +236,7 @@ def run_demo_server(pipe):
     ) as demo:
         gr.Markdown(
             """
-            # StableDelight: Real-Time Reflection Removal from Textured Surfaces
+            # StableDelight: Removing Reflections from Textured Surfaces in a Single Image
             <p align="center">
         """
         )
@@ -271,7 +251,7 @@ def run_demo_server(pipe):
                         )
                         with gr.Row():
                             image_submit_btn = gr.Button(
-                                value="Delighting", variant="primary"
+                                value="Delightning", variant="primary"
                             )
                             image_reset_btn = gr.Button(value="Reset")
                     with gr.Column():