Update app.py

app.py

@@ -1,7 +1,9 @@
 import os
 import spaces
 import gradio as gr
-from detection import EyesDetection
+
+from detection import EyesDetection, FaceDetection, HeadDetection, PersonDetection, HandDetection, CensorDetection, \
+    HalfBodyDetection, NudeNetDetection, BooruYOLODetection
 
 _GLOBAL_CSS = """
 .limit-height {
@@ -9,26 +11,34 @@ _GLOBAL_CSS = """
 }
 """
 
-eyes_detector = EyesDetection()
-
-@spaces.GPU
-def run_detection(img):
-    return eyes_detector.detect(img, model_name="eye_detect_v1.0_s")  # This function must exist in your class
-
 if __name__ == '__main__':
     with gr.Blocks(css=_GLOBAL_CSS) as demo:
         with gr.Row():
             with gr.Column():
                 gr.HTML('<h2 style="text-align: center;">Object Detections For Anime</h2>')
-                gr.Markdown('This is the online demo for detection functions...')
+                gr.Markdown('This is the online demo for detection functions of '
+                            '[imgutils.detect](https://dghs-imgutils.deepghs.org/main/api_doc/detect/index.html). '
+                            'You can try them yourselves with `pip install dghs-imgutils`.')
 
         with gr.Row():
             with gr.Tabs():
+                with gr.Tab('Face Detection'):
+                    FaceDetection().make_ui()
+                with gr.Tab('Head Detection'):
+                    HeadDetection().make_ui()
+                with gr.Tab('Person Detection'):
+                    PersonDetection().make_ui()
+                with gr.Tab('Half Body Detection'):
+                    HalfBodyDetection().make_ui()
                 with gr.Tab('Eyes Detection'):
-                    with gr.Row():
-                        img = gr.Image(type="numpy")
-                        label = gr.Label()
-                        btn = gr.Button("Detect Eyes")
-                        btn.click(run_detection, inputs=img, outputs=label)
+                    EyesDetection().make_ui()
+                with gr.Tab('Hand Detection'):
+                    HandDetection().make_ui()
+                with gr.Tab('Censor Point Detection'):
+                    CensorDetection().make_ui()
+                with gr.Tab('NudeNet'):
+                    NudeNetDetection().make_ui()
+                with gr.Tab('BooruYOLO'):
+                    BooruYOLODetection().make_ui()
 
-    demo.queue(os.cpu_count()).launch()
+    demo.queue(os.cpu_count()).launch()