Working prototype

.gitignore

@@ -0,0 +1 @@
+flagged/

app.py

@@ -1,21 +1,41 @@
-from PIL import UnidentifiedImageError
 import gradio as gr
 
-from nudenet import NudeDetector
+from nudenetupdated import NudeDetector
 
-nudenet_detector = NudeDetector()
+detector = NudeDetector()
+
+
+def predict(image, checkbox):
+    if checkbox:
+        mode = "fast"
+    else:
+        mode = "default"
 
-def predict(image):
     try:
-        r = nudenet_detector.detect(image)
-        return r
+        result = detector.detect(image, mode)
+
+        sections = []
+        for i in range(len(result)):
+            print()
+            sections.append((
+                    result[i]['box'],
+                    result[i]['label']
+            ))
+        
+        output = (image, sections)
+        return (output, result)
     except Exception as e:
         print(f"Got uncaught exception {type(e)}: {e}")
 
+
 gr.Interface(
     fn=predict,
     inputs=[
         gr.Image(),
+        gr.Checkbox(label="Fast"),
     ],
-    outputs=gr.JSON(),
+    outputs=[
+        gr.AnnotatedImage(label="Annotated Image"),
+        gr.JSON(label="JSON")
+    ]
 ).launch(server_name="0.0.0.0")

models/classes

@@ -1,16 +0,0 @@
-EXPOSED_ANUS
-EXPOSED_ARMPITS
-COVERED_BELLY
-EXPOSED_BELLY
-COVERED_BUTTOCKS
-EXPOSED_BUTTOCKS
-FACE_F
-FACE_M
-COVERED_FEET
-EXPOSED_FEET
-COVERED_BREAST_F
-EXPOSED_BREAST_F
-COVERED_GENITALIA_F
-EXPOSED_GENITALIA_F
-EXPOSED_BREAST_M
-EXPOSED_GENITALIA_M

models/detector_v2_default_checkpoint.onnx

@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:d4be1c504be61851d9745e6da8fa09455eb39b8856626dd6b5ca413c9e8b1578
-size 146675691

requirements.txt

@@ -1,2 +1 @@
 gradio
-nudenet==2.0.9