Update app.py
Browse files
app.py
CHANGED
@@ -197,6 +197,7 @@ import streamlit as st
|
|
197 |
from streamlit_webrtc import webrtc_streamer
|
198 |
import av
|
199 |
import queue
|
|
|
200 |
|
201 |
# Logging setup
|
202 |
logger = logging.getLogger(__name__)
|
@@ -220,6 +221,7 @@ def calculate_angle(a, b, c):
|
|
220 |
angle = 360 - angle
|
221 |
return angle
|
222 |
|
|
|
223 |
# Detection Queue
|
224 |
result_queue: queue.Queue[List[Detection]] = queue.Queue()
|
225 |
|
@@ -284,6 +286,7 @@ def video_frame_callback(frame: av.VideoFrame) -> av.VideoFrame:
|
|
284 |
return av.VideoFrame.from_ndarray(image, format="bgr24")
|
285 |
|
286 |
|
|
|
287 |
# WebRTC streamer configuration
|
288 |
webrtc_streamer(
|
289 |
key="squat-detection",
|
|
|
197 |
from streamlit_webrtc import webrtc_streamer
|
198 |
import av
|
199 |
import queue
|
200 |
+
from typing import List
|
201 |
|
202 |
# Logging setup
|
203 |
logger = logging.getLogger(__name__)
|
|
|
221 |
angle = 360 - angle
|
222 |
return angle
|
223 |
|
224 |
+
|
225 |
# Detection Queue
|
226 |
result_queue: queue.Queue[List[Detection]] = queue.Queue()
|
227 |
|
|
|
286 |
return av.VideoFrame.from_ndarray(image, format="bgr24")
|
287 |
|
288 |
|
289 |
+
|
290 |
# WebRTC streamer configuration
|
291 |
webrtc_streamer(
|
292 |
key="squat-detection",
|