Upload app.py

app.py

@@ -4,6 +4,7 @@ import numpy as np
 import torch
 from PIL import Image
 import gradio as gr
+from gradio_imageslider import ImageSlider  # 导入 gradio_imageslider
 
 # 延迟 CUDA 初始化
 weight_dtype = torch.float32
@@ -45,6 +46,17 @@ def process_image(input_image):
     # 将 Gradio 输入转换为 PIL 图像
     input_image = Image.fromarray(input_image)
 
+    # 调整输入图像的最大边为 768
+    width, height = input_image.size
+    max_size = 768
+    if width > height:
+        new_width = max_size
+        new_height = int(height * (max_size / width))
+    else:
+        new_height = max_size
+        new_width = int(width * (max_size / height))
+    resized_input_image = input_image.resize((new_width, new_height), Image.LANCZOS)
+
     # 处理图像
     pipe_out = pipe(
         image=input_image,
@@ -58,7 +70,7 @@ def process_image(input_image):
     processed_frame = (processed_frame[0] * 255).astype(np.uint8)
     processed_frame = Image.fromarray(processed_frame)
 
-    return processed_frame
+    return resized_input_image, processed_frame
 
 # 创建 Gradio 界面
 def create_gradio_interface():
@@ -70,7 +82,6 @@ def create_gradio_interface():
     description = """Official demo for **Dereflection Any Image**.
     Please refer to our [paper](), [project page](https://abuuu122.github.io/DAI.github.io/), and [github](https://github.com/Abuuu122/Dereflection-Any-Image) for more details."""
 
-
     with gr.Blocks() as demo:
         gr.Markdown(title)
         gr.Markdown(description)
@@ -79,7 +90,7 @@ def create_gradio_interface():
                 input_image = gr.Image(label="Input Image", type="numpy")
                 submit_btn = gr.Button("Remove Reflection", variant="primary")
             with gr.Column():
-                output_image = gr.Image(label="Processed Image")
+                output_image = ImageSlider(label="Processed Image")  # 使用 ImageSlider
 
         # 添加示例
         gr.Examples(