Update app.py

app.py

@@ -12,13 +12,13 @@ def yolov8_func(image,
     # Load the YOLOv8 model
     model_path = "best.pt"
     model = YOLO(model_path)  
-
+    
     # Make predictions
     result = model.predict(image, conf=conf_thresold, iou=iou_thresold, imgsz=image_size)
 
     # Access object detection results
     boxes = result[0].boxes  
-    num_boxes = len(boxes)   # Count the number of bounding boxes (detections)
+    num_boxes = len(boxes)   
 
     # Print object detection details (optional)
     print("Object type: ", boxes.cls)
@@ -43,26 +43,27 @@ def yolov8_func(image,
     # Render the result (with bounding boxes/labels)
     render = render_result(model=model, image=image, result=result[0])
     
-    # Save the rendered image (with predictions)
     predicted_image_save_path = "predicted_image.jpg"
     render.save(predicted_image_save_path)
-
-    # Return the saved image, severity, and recommendation for Gradio output
     return predicted_image_save_path, f"Acne condition: {severity}", recommendation
 
 # Define inputs for the Gradio app
 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.25, label="Confidence Threshold"),
-    gr.Slider(minimum=0, maximum=1, step=0.05, value=0.45, label="IOU Threshold")
+    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")
 ]
 
-# Define the output for the Gradio app (image + text for severity and recommendation)
+# Use a Row layout to align the textboxes for condition and recommendation
+output_image = gr.Image(type="filepath", label="Output Image")
+acne_condition = gr.Textbox(label="Acne Condition")
+recommendation = gr.Textbox(label="Recommendation")
+
+# Define the layout using Rows and Columns
 outputs = [
-    gr.Image(type="filepath", label="Output Image"),
-    gr.Textbox(label="Acne Condition"),
-    gr.Textbox(label="Recommendation")
+    output_image,
+    gr.Row([acne_condition, recommendation])
 ]
 
 # Set the title of the Gradio app
@@ -75,4 +76,4 @@ yolo_app = gr.Interface(fn=yolov8_func,
                     title=title)
 
 # Launch the app
-yolo_app.launch(debug=True)
+yolo_app.launch(debug=True)