jatinmehra commited on
Commit
ed0f068
·
verified ·
1 Parent(s): 24416ff

Fix webcam mode AttributeError by using MediaMode enum

Browse files

Fix webcam mode AttributeError by using MediaMode enum

Fixed the AttributeError: 'str' object has no attribute 'name' when switching to webcam mode.
Changed the string value "recv" to use the proper MediaMode.RECV enum from streamlit_webrtc package.
This ensures the mode parameter has the .name attribute that webrtc_streamer() expects.

Files changed (1) hide show
  1. app.py +2 -3
app.py CHANGED
@@ -5,7 +5,7 @@ import numpy as np
5
  import torch
6
  from collections import deque
7
  from transformers import AutoFeatureExtractor, AutoModelForVideoClassification
8
- from streamlit_webrtc import webrtc_streamer, VideoTransformerBase, RTCConfiguration
9
 
10
  # Constants
11
  NUM_FRAMES = 16
@@ -29,7 +29,6 @@ st.title("Dashcam Accident Predictor")
29
  st.write("**higher score = higher accident probability**")
30
 
31
  # Function to run inference on a saved video file
32
-
33
  def run_inference_on_video(video_path):
34
  cap = cv2.VideoCapture(video_path)
35
  total_frames = int(cap.get(cv2.CAP_PROP_FRAME_COUNT))
@@ -96,7 +95,7 @@ else:
96
 
97
  webrtc_streamer(
98
  key="dashcam-webcam",
99
- mode="recv",
100
  rtc_configuration=RTCConfiguration({
101
  "iceServers": [{"urls": ["stun:stun.l.google.com:19302"]}]
102
  }),
 
5
  import torch
6
  from collections import deque
7
  from transformers import AutoFeatureExtractor, AutoModelForVideoClassification
8
+ from streamlit_webrtc import webrtc_streamer, VideoTransformerBase, RTCConfiguration, MediaMode
9
 
10
  # Constants
11
  NUM_FRAMES = 16
 
29
  st.write("**higher score = higher accident probability**")
30
 
31
  # Function to run inference on a saved video file
 
32
  def run_inference_on_video(video_path):
33
  cap = cv2.VideoCapture(video_path)
34
  total_frames = int(cap.get(cv2.CAP_PROP_FRAME_COUNT))
 
95
 
96
  webrtc_streamer(
97
  key="dashcam-webcam",
98
+ mode=MediaMode.RECV, # Changed from "recv" string to MediaMode enum
99
  rtc_configuration=RTCConfiguration({
100
  "iceServers": [{"urls": ["stun:stun.l.google.com:19302"]}]
101
  }),