Spaces:

akhaliq
/

yolov7

Runtime error

akhaliq HF staff commited on Jul 8, 2022

Commit

a9171f2

•

1 Parent(s): d0a2f17

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -180,7 +180,7 @@ def detect(img):
     print(f'Done. ({time.time() - t0:.3f}s)')
-    return Image.fromarray(im0)
 with torch.no_grad():
     if opt.update:  # update all models (to fix SourceChangeWarning)

     print(f'Done. ({time.time() - t0:.3f}s)')
+    return Image.fromarray(im0[:,:,::-1])
 with torch.no_grad():
     if opt.update:  # update all models (to fix SourceChangeWarning)