Update app.py

app.py

@@ -363,17 +363,15 @@ def process_line(img_path='./input', outdir='./output'):
 if __name__ == '__main__':
     # Gradio 接口代码
     import gradio as gr
-    from PIL import Image
 
     def gradio_process_line(img):
-    img_path = './temp_input.png'
-    img.save(img_path)
+        img_path = './temp_input.png'
+        img.save(img_path)
 
-    process_line(img_path, './output')
-
-    output_image_path = './output/teed_imgs/processed_image.png'  # 更新为实际输出路径
-    return Image.open(output_image_path)
+        process_line(img_path, './output')
 
+        output_image_path = './output/teed_imgs/processed_image.png'  # 更新为实际输出路径
+        return Image.open(output_image_path)
 
     # 定义 Gradio 接口
     iface = gr.Interface(
@@ -386,7 +384,6 @@ if __name__ == '__main__':
 
     # 启动 Gradio 应用
     iface.launch()
-