2384603g commited on
Commit
090e4ff
·
verified ·
1 Parent(s): f2b595a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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(result_img, cv2.COLOR_BGR2RGB))
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