Update app.py

app.py

@@ -27,10 +27,10 @@ def yolov8_func(image,
     print(f"Number of bounding boxes: {num_boxes}")
 
     # Categorize based on number of boxes (detections) and provide recommendations
-    if num_boxes > 20:
+    if num_boxes > 10:
         severity = "Worse"
         recommendation = "It is recommended to see a dermatologist and start stronger acne treatment."
-    elif 10 <= num_boxes <= 20:
+    elif 5 <= num_boxes <= 10:
         severity = "Medium"
         recommendation = "You should follow a consistent skincare routine with proper cleansing and moisturizing."
     else:
@@ -55,7 +55,7 @@ inputs = [
     gr.Image(type="filepath", label="Input Image"),
     gr.Slider(minimum=320, maximum=1280, step=32, value=640, label="Image Size"),
     gr.Slider(minimum=0, maximum=1, step=0.05, value=0.15, label="Confidence Threshold"),
-    gr.Slider(minimum=0, maximum=1, step=0.05, value=0.2, label="IOU Threshold")
+    gr.Slider(minimum=0, maximum=1, step=0.05, value=0.15, label="IOU Threshold")
 ]
 
 # Define the output for the Gradio app (image + text for severity and recommendation)