Spaces:

LearningnRunning
/

adult_image_detector

Running

LearningnRunning commited on Sep 3, 2024

Commit

76b6140

1 Parent(s): 489f48b

FEAT confidence threshold

Files changed (2) hide show

__pycache__/app.cpython-310.pyc ADDED Viewed

Binary file (3.6 kB). View file

app.py CHANGED Viewed

@@ -113,7 +113,7 @@ demo = gr.Interface(
     fn=detect_nsfw,
     inputs=[
         gr.Image(type="numpy", label="Upload an image or enter a URL"),
-        gr.Slider(0, 1, value=0.1, label="Confidence Threshold"),
         gr.Slider(0, 1, value=0.45, label="Overlap Threshold"),
         gr.Dropdown(["Draw box", "Draw Label", "Draw Confidence", "Censor Predictions"], label="Label Display Mode", value="Draw box")
     ],

     fn=detect_nsfw,
     inputs=[
         gr.Image(type="numpy", label="Upload an image or enter a URL"),
+        gr.Slider(0, 1, value=0.45, label="Confidence Threshold"),
         gr.Slider(0, 1, value=0.45, label="Overlap Threshold"),
         gr.Dropdown(["Draw box", "Draw Label", "Draw Confidence", "Censor Predictions"], label="Label Display Mode", value="Draw box")
     ],