Update app.py

app.py

@@ -1,44 +1,21 @@
 import os
 import spaces
 import gradio as gr
+from detection import EyesDetection
 
-from detection import EyesDetection, FaceDetection, HeadDetection, PersonDetection, HandDetection, CensorDetection, \
-    HalfBodyDetection, NudeNetDetection, BooruYOLODetection
+eyes_detector = EyesDetection()
 
-_GLOBAL_CSS = """
-.limit-height {
-    max-height: 55vh;
-}
-"""
+# expose the heavy work so ZeroGPU sees it
+@spaces.GPU
+def _gpu_detect(img, m="eye_detect_v1.0_s"):
+    return eyes_detector.detect(img, model_name=m)
 
-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 of '
-                            '[imgutils.detect](https://dghs-imgutils.deepghs.org/main/api_doc/detect/index.html). '
-                            'You can try them yourselves with `pip install dghs-imgutils`.')
+if __name__ == "__main__":
+    with gr.Blocks() as demo:
+        with gr.Tabs():
+            with gr.Tab("Eyes Detection"):
+                # the helper builds an image uploader, button,
+                # and nicely formatted results table
+                eyes_detector.make_ui(detect_fn=_gpu_detect)
 
-        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'):
-                    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(share=True)
+    demo.queue(os.cpu_count()).launch()