Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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")
|