Modify state initialization

app.py

@@ -163,10 +163,30 @@ class RunningResult:
         return self.predictions[-1][1] if self.predictions else "Starting..."
 
 
-def process_frames(image: np.ndarray, running_frames_cache: RunningFramesCache, running_result: RunningResult):
+def process_frames(image: np.ndarray, frames_state: list, result_state: list):
+    
+    # Initialize frames cache if not exists (and put in gradio state)
+    if not frames_state:
+        running_frames_cache = RunningFramesCache(
+            save_every_k_frame=128 / frames_per_clip,
+            max_frames=frames_per_clip,
+        )
+        frames_state.append(running_frames_cache)
+    else:
+        running_frames_cache = frames_state[0]
+
+    # Initialize result cache if not exists (and put in gradio state)
+    if not result_state:
+        running_result = RunningResult(4)
+        result_state.append(running_result)
+    else:
+        running_result = result_state[0]
+
+    # Add frame to frames cache
     image = np.flip(image, axis=1).copy()
     running_frames_cache.add_frame(image)
 
+    # Run model if enough frames are available
     if (
         running_frames_cache.counter % UPDATE_EVERY_N_FRAMES == 0
         and len(running_frames_cache) >= model.config.frames_per_clip
@@ -187,6 +207,7 @@ def process_frames(image: np.ndarray, running_frames_cache: RunningFramesCache,
         class_name = model.config.id2label[top_index]
         running_result.add_prediction(class_name)
 
+    # Get formatted predictions and last prediction
     formatted_predictions = running_result.get_formatted_predictions()
     last_prediction = running_result.get_last_prediction()
     image = add_text_on_image(image, last_prediction)
@@ -197,13 +218,8 @@ async def get_credentials():
     return await get_cloudflare_turn_credentials_async(hf_token=HF_TOKEN)
 
 
-frames_cache = gr.State(
-    RunningFramesCache(
-        save_every_k_frame=128 / frames_per_clip,
-        max_frames=frames_per_clip,
-    )
-)
-result_cache = gr.State(RunningResult(4))
+frames_cache = gr.State([])
+result_cache = gr.State([])
 
 # Initialize the video stream with processing callback
 stream = Stream(