Upload monochrome_detection/app.py with huggingface_hub

monochrome_detection/app.py

@@ -0,0 +1,32 @@
+import os
+from typing import Mapping
+
+import gradio as gr
+from PIL import Image
+
+from monochrome import _DEFAULT_MONOCHROME_CKPT, _MONOCHROME_CKPTS, get_monochrome_score
+
+
+def monochrome_detect(image: Image.Image, ckpt: str) -> Mapping[str, float]:
+    return get_monochrome_score(image, ckpt)
+
+
+if __name__ == '__main__':
+    with gr.Blocks() as demo:
+        with gr.Row():
+            with gr.Column():
+                gr_input_image = gr.Image(type='pil', label='Original Image')
+                with gr.Row():
+                    gr_model = gr.Dropdown(_MONOCHROME_CKPTS, value=_DEFAULT_MONOCHROME_CKPT, label='Model')
+
+                gr_btn_submit = gr.Button(value='Detection', variant='primary')
+
+            with gr.Column():
+                gr_tags = gr.Label(label='Detection')
+
+        gr_btn_submit.click(
+            monochrome_detect,
+            inputs=[gr_input_image, gr_model],
+            outputs=[gr_tags],
+        )
+    demo.queue(os.cpu_count()).launch()