Optimize image process

app.py

@@ -1,4 +1,5 @@
 import tempfile
+import cv2
 import gradio as gr
 import tensorflow as tf
 from moviepy.editor import VideoFileClip
@@ -39,7 +40,11 @@ def fn(video: gr.Video):
       frame = draw_boxes(frame, detections, actions)
       processed_frames.append(frame)
       if i % config.yield_frame_steps == 0:
-        yield frame, None
+        with tempfile.NamedTemporaryFile(suffix='.jpeg') as f:
+          quality = 1
+          _, img_encoded = cv2.imencode('.jpg', img_array, [int(cv2.IMWRITE_JPEG_QUALITY), quality])
+          f.write(img_encoded)
+          yield f.name, None
     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)

configuration.py

@@ -29,4 +29,4 @@ class Config:
   detect_object_frame_steps = 5
   classify_action_frame_steps = 15
   classify_action_num_frames = 8
-  yield_frame_steps = 1
+  yield_frame_steps = 5