import os | |
import gradio as gr | |
from base import Classification | |
apps = [ | |
Classification('Classification', 'deepghs/anime_classification', 'mobilenetv3_sce_dist'), | |
Classification('Monochrome', 'deepghs/monochrome_detect', 'mobilenetv3_large_100_dist'), | |
Classification('AI Check', 'deepghs/anime_ai_check', 'mobilenetv3_sce_dist'), | |
Classification('Rating', 'deepghs/anime_rating', 'mobilenetv3_sce_dist'), | |
Classification('Character Sex', 'deepghs/anime_ch_sex', 'caformer_s36_v1'), | |
Classification('Character Skin', 'deepghs/anime_ch_skin_color', 'caformer_s36'), | |
] | |
if __name__ == '__main__': | |
with gr.Blocks() as demo: | |
with gr.Tabs(): | |
for cls in apps: | |
cls.create_gr() | |
demo.queue(os.cpu_count()).launch() | |