File size: 2,030 Bytes
2023a9f e155f7f 3255ddb e155f7f 3255ddb e155f7f fb02dfc e155f7f a5dfe9a fb02dfc 3255ddb e2a3bbe e155f7f 4912df2 64b80f6 663c67b 2f68780 e155f7f 2023a9f e155f7f 21e723b 2023a9f |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 |
import os
import gradio as gr
from base import Classification
apps = [
Classification('Classification', 'deepghs/anime_classification', 'mobilenetv3_v1.3_dist'),
Classification('Monochrome', 'deepghs/monochrome_detect', 'mobilenetv3_large_100_dist'),
Classification('Completeness', 'deepghs/anime_completeness', 'mobilenetv3_v2.2_dist'),
Classification('AI Check', 'deepghs/anime_ai_check', 'mobilenetv3_sce_dist'),
Classification('AI Corrupt', 'deepghs/ai_image_corrupted', 'caformer_s36_v0_focal'),
Classification('Rating', 'deepghs/anime_rating', 'mobilenetv3_sce_dist'),
#Classification('Character Sex', 'deepghs/anime_ch_sex', 'caformer_s36_v1'),
Classification('Portrait Type', 'deepghs/anime_portrait', 'mobilenetv3_v0_dist'),
Classification('Age of Style', 'deepghs/anime_style_ages', 'mobilenetv3_v0_dist'),
Classification('Bangumi Portrait', 'deepghs/bangumi_char_type', 'mobilenetv3_v0_dist'),
Classification('Is That Anime?', 'deepghs/anime_real_cls', 'mobilenetv3_v1.2_dist'),
Classification('Teen', 'deepghs/anime_teen', 'caformer_s36_v0'),
Classification('Character Skin', 'deepghs/anime_ch_skin_color', 'caformer_s36'),
Classification('Character Hair Color', 'deepghs/anime_ch_hair_color', 'caformer_s36_v0'),
Classification('Character Eye Color', 'deepghs/anime_ch_eye_color', 'caformer_s36_v0'),
Classification('Character Hair Length', 'deepghs/anime_ch_hair_length', 'caformer_s36_v0'),
Classification('Character Ears', 'deepghs/anime_ch_ear', 'caformer_s36_v0'),
Classification('Character Horns', 'deepghs/anime_ch_horn', 'caformer_s36_v0'),
Classification('[Beta] Danbooru Rating', 'deepghs/anime_dbrating', 'mobilenetv3_large_100_v0_ls0.2'),
Classification('[Beta] Danbooru Aesthetic', 'deepghs/anime_aesthetic', 'caformer_s36_v0_ls0.2'),
]
if __name__ == '__main__':
with gr.Blocks() as demo:
with gr.Tabs():
for cls in apps:
cls.create_gr()
demo.queue(os.cpu_count()).launch()
|