luminoussg commited on
Commit
edf1813
·
verified ·
1 Parent(s): 3e906aa

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -4
app.py CHANGED
@@ -23,7 +23,7 @@ def resize_frame(frame, scale_percent):
23
  return resized
24
 
25
  @spaces.GPU
26
- def process_video(video_file, scale_percent, line_start_x, line_start_y, line_end_x, line_end_y, line_thickness, draw_tracks, view_img, view_in_counts, view_out_counts, track_thickness, region_thickness, line_dist_thresh, persist, conf, iou, classes, verbose, progress=gr.Progress()):
27
  # Ensure classes is a list of integers
28
  classes = [int(x) for x in classes.split(',') if x.strip().isdigit()] if classes else None
29
 
@@ -93,7 +93,7 @@ def process_video(video_file, scale_percent, line_start_x, line_start_y, line_en
93
 
94
  # Update progress
95
  processed_frames += 1
96
- progress.update(processed_frames / total_frames)
97
 
98
  cap.release()
99
  video_writer.release()
@@ -129,8 +129,7 @@ def preview_line(video_file, scale_percent, line_start_x, line_start_y, line_end
129
  return preview_path
130
 
131
  def gradio_app(video, scale_percent, line_start_x, line_start_y, line_end_x, line_end_y, line_thickness, draw_tracks, view_img, view_in_counts, view_out_counts, track_thickness, region_thickness, line_dist_thresh, persist, conf, iou, classes_to_track, verbose):
132
- progress = gr.Progress()
133
- output_path = process_video(video.name, scale_percent, line_start_x, line_start_y, line_end_x, line_end_y, int(line_thickness), draw_tracks, view_img, view_in_counts, view_out_counts, int(track_thickness), int(region_thickness), line_dist_thresh, persist, conf, iou, classes_to_track, verbose, progress)
134
  return output_path, output_path
135
 
136
  def update_preview(video, scale_percent, line_start_x, line_start_y, line_end_x, line_end_y, line_thickness):
 
23
  return resized
24
 
25
  @spaces.GPU
26
+ def process_video(video_file, scale_percent, line_start_x, line_start_y, line_end_x, line_end_y, line_thickness, draw_tracks, view_img, view_in_counts, view_out_counts, track_thickness, region_thickness, line_dist_thresh, persist, conf, iou, classes, verbose, progress):
27
  # Ensure classes is a list of integers
28
  classes = [int(x) for x in classes.split(',') if x.strip().isdigit()] if classes else None
29
 
 
93
 
94
  # Update progress
95
  processed_frames += 1
96
+ progress((processed_frames / total_frames) * 100)
97
 
98
  cap.release()
99
  video_writer.release()
 
129
  return preview_path
130
 
131
  def gradio_app(video, scale_percent, line_start_x, line_start_y, line_end_x, line_end_y, line_thickness, draw_tracks, view_img, view_in_counts, view_out_counts, track_thickness, region_thickness, line_dist_thresh, persist, conf, iou, classes_to_track, verbose):
132
+ output_path = process_video(video.name, scale_percent, line_start_x, line_start_y, line_end_x, line_end_y, int(line_thickness), draw_tracks, view_img, view_in_counts, view_out_counts, int(track_thickness), int(region_thickness), line_dist_thresh, persist, conf, iou, classes_to_track, verbose, gr.Progress())
 
133
  return output_path, output_path
134
 
135
  def update_preview(video, scale_percent, line_start_x, line_start_y, line_end_x, line_end_y, line_thickness):