BrainAI-1 commited on
Commit
07e7302
·
verified ·
1 Parent(s): bf01437

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -7
app.py CHANGED
@@ -12,10 +12,10 @@ st.title("🎥 YOLO Object Detection Live Stream")
12
 
13
 
14
  image = camera_input_live()
15
-
16
- uploaded_image = PIL.Image.open(image)
17
- #uploaded_image_cv = cv2.cvtColor(np.array(uploaded_image), cv2.COLOR_RGB2BGR)
18
- result = model(uploaded_image)
19
- img_plot = result[0].plot()
20
-
21
- st.image(img_plot, channels = "BGR")
 
12
 
13
 
14
  image = camera_input_live()
15
+ if image:
16
+ uploaded_image = PIL.Image.open(image)
17
+ #uploaded_image_cv = cv2.cvtColor(np.array(uploaded_image), cv2.COLOR_RGB2BGR)
18
+ result = model(uploaded_image)
19
+ img_plot = result[0].plot()
20
+
21
+ st.image(img_plot, channels = "BGR")