to-be commited on
Commit
6ac1620
1 Parent(s): 196a89d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -62,7 +62,7 @@ def inference(img, task, run_on):
62
  torch.cuda.empty_cache()
63
 
64
  open_cv_image = np.array(img)
65
- img = cv2.cvtColor(cv2.imread(open_cv_image), cv2.COLOR_BGR2RGB)
66
 
67
  input_ = torch.from_numpy(img).float().div(255.).permute(2,0,1).unsqueeze(0).to(device)
68
 
 
62
  torch.cuda.empty_cache()
63
 
64
  open_cv_image = np.array(img)
65
+ img = cv2.cvtColor(open_cv_image, cv2.COLOR_BGR2RGB)
66
 
67
  input_ = torch.from_numpy(img).float().div(255.).permute(2,0,1).unsqueeze(0).to(device)
68