Piarasingh85 commited on
Commit
1a6b50c
·
verified ·
1 Parent(s): 8e54490

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -3
app.py CHANGED
@@ -2,12 +2,16 @@ import gradio as gr
2
  import cv2
3
 
4
  def video_feed():
5
- cap = cv2.VideoCapture(0) # Use 0 for default camera
 
 
6
  while True:
7
  ret, frame = cap.read()
8
  if not ret:
9
- break
10
- yield cv2.imencode('.jpg', frame)[1].tobytes()
 
 
11
 
12
  cap.release()
13
 
 
2
  import cv2
3
 
4
  def video_feed():
5
+ cap = cv2.VideoCapture(0) # Use 0 for the default camera
6
+ if not cap.isOpened():
7
+ raise RuntimeError("Could not start camera.")
8
  while True:
9
  ret, frame = cap.read()
10
  if not ret:
11
+ continue
12
+ _, buffer = cv2.imencode('.jpg', frame)
13
+ frame = buffer.tobytes()
14
+ yield frame
15
 
16
  cap.release()
17