updated to new api

app.py

@@ -3,7 +3,8 @@ import streamlit as st
 
 from pathlib import Path
 from landingai.predict import Predictor
-from landingai.vision_pipeline import NetworkedCamera, FrameSet
+from landingai.pipeline.image_source import NetworkedCamera, FrameSet
+from landingai.pipeline.postprocessing import get_class_counts
 from landingai.st_utils import render_svg
 
 
@@ -37,11 +38,12 @@ def get_latest_surfer_count():
     for i, frame in enumerate(vid_src):
         if i >= VIDEO_LEN_SEC * FPS:
             break
-        frs.extend(frame.run_predict(predictor=surfer_model).overlay_predictions())
+        frs.extend(frame)
+    frs.run_predict(predictor=surfer_model).overlay_predictions()
     frs.save_video(str(VIDEO_CACHE_PATH), video_fps=FPS, image_src="overlay")
-    counts = frs.get_class_counts()
+    counts = get_class_counts(frs)
     if "surfer" in counts:
-        surfers = frs.get_class_counts()["surfer"] / (VIDEO_LEN_SEC * FPS)
+        surfers = counts["surfer"] / (VIDEO_LEN_SEC * FPS)
     else:
         surfers = 0
     st.video(open(VIDEO_CACHE_PATH, "rb").read())