Upload app.py

app.py

@@ -4,7 +4,6 @@ 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
@@ -46,17 +45,6 @@ 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,
@@ -70,7 +58,7 @@ def process_image(input_image):
     processed_frame = (processed_frame[0] * 255).astype(np.uint8)
     processed_frame = Image.fromarray(processed_frame)
 
-    return resized_input_image, processed_frame
+    return processed_frame
 
 # 创建 Gradio 界面
 def create_gradio_interface():
@@ -82,6 +70,7 @@ 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)
@@ -90,7 +79,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 = ImageSlider(label="Processed Image")  # 使用 ImageSlider
+                output_image = gr.Image(label="Processed Image")
 
         # 添加示例
         gr.Examples(