Pratyush101 commited on
Commit
c324fea
·
verified ·
1 Parent(s): 1c15637

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -0
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",