Update src/yolo/predict_pose.py

src/yolo/predict_pose.py

@@ -2,14 +2,12 @@ import sys
 import PIL.Image as Image
 from ultralytics import YOLO
 import gradio as gr
-# import spaces
 
 # Local imports
 from src.logger import logging
 from src.exception import CustomExceptionHandling
 
 
-# @spaces.GPU
 def predict_pose(
     img: str,
     conf_threshold: float,
@@ -25,7 +23,7 @@ def predict_pose(
         - conf_threshold (float): The confidence threshold for object detection.
         - iou_threshold (float): The Intersection Over Union (IOU) threshold for non-max suppression.
         - max_detections (int): The maximum number of detections allowed.
-        - model_name (str): The name or path of the YOLOv8 model to be used for prediction.
+        - model_name (str): The name or path of the YOLO11 model to be used for prediction.
 
     Returns:
         PIL.Image.Image: The image with predicted objects plotted on it.