Update app.py

app.py

@@ -1,7 +1,6 @@
 import gradio as gr
 import cv2
 import numpy as np
-from skimage import io, color, restoration
 
 # 圖像分割功能
 def image_segmentation(image):
@@ -21,12 +20,6 @@ def image_inpainting(image, mask):
     inpainted = cv2.inpaint(image, mask, inpaintRadius=3, flags=cv2.INPAINT_TELEA)
     return inpainted
 
-# 示例圖片列表
-example_images = [
-    ("example1.jpg", "測試圖片 1"),
-    ("example2.jpg", "測試圖片 2"),
-]
-
 # 定義 Gradio 接口
 def main_interface(image, task, param1=50, param2=150, mask=None):
     if task == "Image Segmentation":
@@ -47,7 +40,7 @@ with gr.Blocks() as app:
             task = gr.Radio(["Image Segmentation", "Edge Detection", "Image Inpainting"], label="選擇功能")
             param1 = gr.Slider(0, 200, value=50, step=1, label="參數 1 (Edge Detection Threshold 1)")
             param2 = gr.Slider(0, 200, value=150, step=1, label="參數 2 (Edge Detection Threshold 2)")
-            mask_input = gr.Image(label="上傳遮罩 (僅適用於 Image Inpainting)", type="numpy", optional=True)
+            mask_input = gr.Image(label="上傳遮罩 (僅適用於 Image Inpainting)", type="numpy", value=None)
             submit_button = gr.Button("執行")
         with gr.Column():
             output_image = gr.Image(label="輸出結果")