import os

import gradio as gr

from base import Classification

apps = [
    Classification('Completeness [Pro]', 'DeepBase/anime_completeness', 'caformer_s36_v2.3'),
]

if __name__ == '__main__':
    with gr.Blocks() as demo:
        gr.Markdown('''
            **This space is for the PRO models we trained.**

            We will not open its dataset and model files because the data provider of the model has not granted open source permission. Only an online demo can be provided here.
        ''')
        with gr.Tabs():
            for cls in apps:
                cls.create_gr()

    demo.queue(os.cpu_count()).launch()