Update app.py

app.py

@@ -1,17 +1,18 @@
 import numpy as np
 import gradio as gr
 import cv2
+from black import to_black
 
-def to_black(image, transfer_style):
-    if transfer_style == "Hayao":
-         output = cv2.cvtColor(image, cv2.COLOR_BGR2GRAY)# 转换为灰度图像
-         return output
-    elif transfer_style == "Shinkai":
-        return image
-    elif transfer_style == "Kon Satoshi":
-        return image
-    else:
-        return image
+# def to_black(image, transfer_style):
+#     if transfer_style == "Hayao":
+#          output = cv2.cvtColor(image, cv2.COLOR_BGR2GRAY)# 转换为灰度图像
+#          return output
+#     elif transfer_style == "Shinkai":
+#         return image
+#     elif transfer_style == "Kon Satoshi":
+#         return image
+#     else:
+#         return image
 
 def clear_output(input_widget):
     input_widget = np.array([])