Create app.py

app.py

@@ -0,0 +1,42 @@
+import os
+
+import gradio as gr
+from imgutils.generic import YOLOModel
+
+_GLOBAL_CSS = """
+.limit-height {
+    max-height: 55vh;
+}
+"""
+
+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 Real Photos</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`.')
+
+        with gr.Row():
+            with gr.Tabs():
+                with gr.Tab('Face Detection'):
+                    YOLOModel('deepghs/real_face_detection').make_ui(
+                        default_model_name='face_detect_v0_s_yv11',
+                        default_conf_threshold=0.25,
+                    )
+                with gr.Tab('Face Detection (Real Only)'):
+                    YOLOModel('deepghs/yolo-face').make_ui(
+                        default_model_name='yolov10s-face',   
+                    )
+                # with gr.Tab('Head Detection'):
+                #     HeadDetection().make_ui()
+                # with gr.Tab('Person Detection'):
+                #     PersonDetection().make_ui()
+                with gr.Tab('Generic'):
+                    YOLOModel('deepghs/yolos').make_ui(
+                        default_model_name='yolo11s',
+                    )
+               
+
+    demo.queue(os.cpu_count()).launch()