CosmickVisions commited on
Commit
42284b9
·
verified ·
1 Parent(s): b9a4f2a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -2425,7 +2425,7 @@ def main():
2425
  yolo_classes = None # Detect all classes
2426
 
2427
  # Pass this to your processing function in the processing_params
2428
- processing_params["yolo_classes"] = yolo_classes
2429
 
2430
  # Process the video with the parameters
2431
  processed_video, results = process_video_file(uploaded_file, analysis_types, **processing_params)
 
2425
  yolo_classes = None # Detect all classes
2426
 
2427
  # Pass this to your processing function in the processing_params
2428
+ processing_params["enabled_classes"] = yolo_classes
2429
 
2430
  # Process the video with the parameters
2431
  processed_video, results = process_video_file(uploaded_file, analysis_types, **processing_params)