Tidy

app.py

@@ -25,6 +25,7 @@ def fn(video: gr.Video, actions: list[int]):
   print('Process video.')
   do_detect = 0 in actions
   do_classify = 1 in actions
+  toggle = True
   if not do_detect and not do_classify:
     yield None, video
     return
@@ -60,7 +61,11 @@ def fn(video: gr.Video, actions: list[int]):
         with tempfile.NamedTemporaryFile(suffix='.jpeg') as f:
           f.write(image_encoded)
           yield f.name, None
-          time.sleep(0.4)
+          if toggle:
+            time.sleep(2)
+            toggle = False
+      if i % config.classify_action_frame_step != 0 and i % config.detect_object_frame_step != 0:
+          time.sleep(0.3)
     processed_clip = ImageSequenceClip(processed_frames, clip.fps)
     processed_clip.audio = clip.audio
     processed_clip.write_videofile(output, fps=clip.fps, audio_codec='aac', logger=None)