SkalskiP commited on
Commit
41975a2
·
1 Parent(s): 8a7385d

drop progress

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -1,6 +1,7 @@
1
  import os
2
  from typing import TypeVar
3
 
 
4
  import gradio as gr
5
  import numpy as np
6
  import supervision as sv
@@ -99,8 +100,7 @@ def video_processing_inference(
99
  input_video: str,
100
  confidence: float,
101
  resolution: int,
102
- checkpoint: str,
103
- progress=gr.Progress()
104
  ):
105
  model = load_model(resolution=resolution, checkpoint=checkpoint)
106
 
@@ -115,7 +115,7 @@ def video_processing_inference(
115
  frames_generator = sv.get_video_frames_generator(input_video, end=total)
116
 
117
  with sv.VideoSink(output_video, video_info=video_info) as sink:
118
- for frame in progress.tqdm(frames_generator, total=total):
119
  annotated_frame = detect_and_annotate(
120
  model=model,
121
  image=frame,
 
1
  import os
2
  from typing import TypeVar
3
 
4
+ from tqdm import tqdm
5
  import gradio as gr
6
  import numpy as np
7
  import supervision as sv
 
100
  input_video: str,
101
  confidence: float,
102
  resolution: int,
103
+ checkpoint: str
 
104
  ):
105
  model = load_model(resolution=resolution, checkpoint=checkpoint)
106
 
 
115
  frames_generator = sv.get_video_frames_generator(input_video, end=total)
116
 
117
  with sv.VideoSink(output_video, video_info=video_info) as sink:
118
+ for frame in tqdm(frames_generator, total=total):
119
  annotated_frame = detect_and_annotate(
120
  model=model,
121
  image=frame,