Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -24,7 +24,7 @@ def process_image(pilimg):
|
|
24 |
# print(x.shape)
|
25 |
result = detection_model.predict(source, conf=0.5)
|
26 |
img_bgr = result[0].plot()
|
27 |
-
return Image.fromarray(cv2.cvtColor(
|
28 |
|
29 |
def process_video(video):
|
30 |
# Open the video file
|
|
|
24 |
# print(x.shape)
|
25 |
result = detection_model.predict(source, conf=0.5)
|
26 |
img_bgr = result[0].plot()
|
27 |
+
return Image.fromarray(cv2.cvtColor(img_bgr, cv2.COLOR_BGR2RGB))
|
28 |
|
29 |
def process_video(video):
|
30 |
# Open the video file
|