Trying to fix a runtime error

app.py

@@ -61,6 +61,8 @@ def detect_objects(model_name,url_input,image_input,threshold):
     feature_extractor = AutoFeatureExtractor.from_pretrained(model_name)
 
     if 'yolov8' in model_name:
+        # Working on getting this to work, another approach
+        # https://docs.ultralytics.com/modes/predict/#key-features-of-predict-mode 
 
         model = YOLO(model_name)
         # set model parameters
@@ -74,14 +76,6 @@ def detect_objects(model_name,url_input,image_input,threshold):
         render = render_result(model=model, image=image_input, result=results[0])
 
         return render, ""
-        
-        # for result in results:
-        #     # https://docs.ultralytics.com/modes/predict/#key-features-of-predict-mode 
-        #     #TODO
-        #     im_array = result.plot()
-        #     im = Image.fromarray(im_array[..., ::=1])
-            
-    
     else:
         if 'detr' in model_name: