nathanjc commited on
Commit
b39bf78
·
1 Parent(s): 950e088

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -152,8 +152,8 @@ def detect(opt, save_img=False):
152
  # Save results (image with detections)
153
  if save_img:
154
  if dataset.mode == 'image':
155
- Image.fromarray(im0).show()
156
- im0.save(save_path)
157
  # cv2.imwrite(save_path, im0)
158
  print(f" The image with the result is saved in: {save_path}")
159
  # else: # 'video' or 'stream'
 
152
  # Save results (image with detections)
153
  if save_img:
154
  if dataset.mode == 'image':
155
+ # Image.fromarray(im0).show()
156
+ Image.fromarray(im0).save(save_path)
157
  # cv2.imwrite(save_path, im0)
158
  print(f" The image with the result is saved in: {save_path}")
159
  # else: # 'video' or 'stream'