Update app.py

app.py

@@ -26,10 +26,16 @@ def inpaint_image(image, mask_threshold):
     # 根據遮罩閾值生成遮罩
     _, mask = cv2.threshold(gray_image, mask_threshold, 255, cv2.THRESH_BINARY)
     mask = mask.astype(bool)
+    
+    # 檢查遮罩是否正確
+    if not mask.any():
+        raise ValueError("遮罩生成失敗，請調整遮罩閾值參數。")
+    
     # 使用 biharmonic 方法修復圖像
     inpainted = inpaint.inpaint_biharmonic(image, mask, multichannel=True)
     return inpainted
 
+
 # 主應用程式
 def app():
     with gr.Blocks() as demo: