Update app.py

app.py

@@ -112,8 +112,10 @@ def yolov8_inference(
         #render = render.resize((dest_width,dest_height))
         #renders.append(render)
         #colorarray.append(colors)
-
-    return model.predict(image)[0].plot()," "
+        renders.append(image_results)
+        num_masks = image_results.masks.data.shape[0]
+        
+    return renders[0].plot(), "Total num: "+str(num_masks)
 
 inputs = [
     gr.Image(type="filepath", label="Input Image"),