Please help me to fix this problem about my code. The problem is the counting object, thats the part of the problem please help me out. Thank you

heres the code:

from flask import Flask, render$\_$template, request, url$\_$for

from werkzeug.utils import secure$\_$filename

import os

from ultralytics import YOLO

import cv$2$

import uuid

app $=$ Flask$(\_\_$name$\_\_)$

app.config$[\text{'}$UPLOAD$\_$FOLDER'$]=$ 'static$/$uploads$\text{'}$

@app.route$(\text{'}/\text{'})$

def index$()$:

return render$\_$template$(\text{'}$uploadindex$.$html$\text{'})$

@app.route$(\text{'}/\text{'},$ methods$=[\text{'}$POST$\text{'}])$

def upload$\_$file$()$:

if 'file' not in request.files:

return $\text{'}$No file part'

file $=$ request.files$[\text{'}$file$\text{'}]$

if file.filename $==\text{'}\text{'}$:

return $\text{'}$No selected file'

if file:

filename $=$ secure$\_$filename$($file$.$filename$)$

file.save$($os$.$path.join$($app$.$config$[\text{'}$UPLOAD$\_$FOLDER'$],$ filename$))$

unique$\_$output$\_$filename $=$ 'uploads$/\text{'}+$ str$($uuid$.$uuid$4())+\text{'}\_$processed.mp$4\text{'}$

class$\_$counts $=$ process$\_$video$($os$.$path.join$($app$.$config$[\text{'}$UPLOAD$\_$FOLDER'$],$ filename$),$ unique$\_$output$\_$filename$)$ # Capture the class counts

return render$\_$template$(\text{'}$video$.$html$\text{'},$ filename$=$unique$\_$output$\_$filename, class$\_$counts$=$class$\_$counts$)$

def process$\_$video$($video$\_$path, output$\_$path$)$:

model $=$ YOLO$(\text{'}$test$6.$pt$\text{'})$

cap $=$ cv$2.$VideoCapture$($video$\_$path$)$

fps $=$ cap.get$($cv$2.$CAP$\_$PROP$\_$FPS$)$

fourcc $=$ cv$2.$VideoWriter$\_$fourcc$(*\text{'}$avc$1\text{'})$ # use 'XVID' codec

out $=$ cv$2.$VideoWriter$(\text{'}$static$/\text{'}+$ output$\_$path, fourcc, fps$,(640,480))$

ret $=$ True

class$1\_$count $=0$ # Initialize class $1$ count

class$2\_$count $=0$ # Initialize class $2$ count

while ret:

ret, frame $=$ cap.read$()$

if ret:

frame $=$ cv$2.$resize$($frame$,(640,480))$ # resize the frame

results $=$ model.track$($frame$,$ persist$=$True, tracker$=$"bytetrack.yaml"$)[0]$

# Count objects for each class

for result in results.boxes.data:

class$\_$id $=$ int$($result$[5])$ # Assuming the class ID is at index $5$

if class$\_$id $==0$: # Replace $0$ with the class ID for class $1$

class$1\_$count $+=1$

elif class$\_$id $==1$: # Replace $1$ with the class ID for class $2$

class$2\_$count $+=1$

frame$\_=$ results.plot$()$

out.write$($frame$\_)$

cap.release$()$

out.release$()$

cv$2.$destroyAllWindows$()$

return $\{\text{'}$class$1\text{'}$: class$1\_$count, 'class$2\text{'}$: class$2\_$count$\}$ # Return the counts of each class

if $\_\_$name$\_\_==\text{'}\_\_$main$\_\_\text{'}$:

app.run$($debug$=$True$)$