Update app.py

app.py

@@ -99,7 +99,7 @@ def yolov8_inference(
         current_model_name = model_name
     model.overrides["conf"] = conf_threshold
     model.overrides["iou"] = iou_threshold
-    model.overrides["classes"] = [0]
+    #model.overrides["classes"] = [0]
     results = model.predict(image)
     renders = []
     colorarray = []
@@ -108,12 +108,12 @@ def yolov8_inference(
         #render = render_result(
         #    model=model, image=image, result=image_results
         #)
-        render ,colors= visualize_masks(image_results.masks.data)
-        render = render.resize((dest_width,dest_height))
-        renders.append(render)
-        colorarray.append(colors)
+        #render ,colors= visualize_masks(image_results.masks.data)
+        #render = render.resize((dest_width,dest_height))
+        #renders.append(render)
+        #colorarray.append(colors)
 
-    return model.predict(image)[0].plot(),','.join(['#%02x%02x%02x' % row for row in colorarray[0]])
+    return model.predict(image)[0].plot()," "
 
 inputs = [
     gr.Image(type="filepath", label="Input Image"),