test

app.py

@@ -13,8 +13,6 @@ import numpy as np
 import PIL.Image
 import tensorflow as tf
 
-DESCRIPTION = '# [KichangKim/DeepDanbooru](https://github.com/KichangKim/DeepDanbooru)'
-
 
 def load_sample_image_paths() -> list[pathlib.Path]:
     image_dir = pathlib.Path('images')
@@ -77,8 +75,10 @@ def predict(
     result_threshold = dict()
     for index in indices:
         label = labels[index]
+        print(label)
         prob = probs[index]
         if label in skip:
+            print("skip", label)
             continue
         if label in translate:
             label = translate[label]
@@ -95,7 +95,6 @@ image_paths = load_sample_image_paths()
 examples = [[path.as_posix(), 0.5] for path in image_paths]
 
 with gr.Blocks(css='style.css') as demo:
-    gr.Markdown(DESCRIPTION)
     with gr.Row():
         with gr.Column():
             image = gr.Image(label='Input', type='pil')