Update application file

app.py

@@ -29,38 +29,6 @@ def process_XDoG(image_path):
     binarized_image = binarize_image(xdog_image)
     return Image.fromarray(binarized_image)
 
-# 主要な色を取得する関数
-def get_major_colors(image, threshold_percentage=0.01):
-    if image.mode != 'RGB':
-        image = image.convert('RGB')
-    color_count = defaultdict(int)
-    for pixel in image.getdata():
-        color_count[pixel] += 1
-    total_pixels = image.width * image.height
-    major_colors = [(color, count) for color, count in color_count.items() if (count / total_pixels) >= threshold_percentage]
-    return major_colors
-
-# 色を統合する関数
-def consolidate_colors(major_colors, threshold):
-    colors_lab = [rgb2lab(np.array([[color]], dtype=np.float32)/255.0).reshape(3) for color, _ in major_colors]
-    i = 0
-    while i < len(colors_lab):
-        j = i + 1
-        while j < len(colors_lab):
-            if deltaE_ciede2000(colors_lab[i], colors_lab[j]) < threshold:
-                if major_colors[i][1] >= major_colors[j][1]:
-                    major_colors[i] = (major_colors[i][0], major_colors[i][1] + major_colors[j][1])
-                    major_colors.pop(j)
-                    colors_lab.pop(j)
-                else:
-                    major_colors[j] = (major_colors[j][0], major_colors[j][1] + major_colors[i][1])
-                    major_colors.pop(i)
-                    colors_lab.pop(i)
-                continue
-            j += 1
-        i += 1
-    return major_colors
-
 # Gradioインターフェース用のメイン関数
 def gradio_interface(image):
     image_path = 'temp_input_image.jpg'
@@ -72,8 +40,8 @@ def gradio_interface(image):
 # Gradioアプリを設定し、起動する
 iface = gr.Interface(
     fn=gradio_interface,
-    inputs=gr.inputs.Image(type='pil', label="Original Image"),
-    outputs=gr.outputs.Image(type='pil', label="Processed Image"),
+    inputs=gr.Image(type='pil', label="Original Image"),
+    outputs=gr.Image(type='pil', label="Processed Image"),
     title="Line Art Removal",
     description="画像をアップロードして線画を除去します。"
 )