youl commited on
Commit
b0035fe
·
1 Parent(s): 717fe15

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -4,6 +4,7 @@ import cv2
4
  import os
5
  import torch.nn as nn
6
  import numpy as np
 
7
  from torchvision.ops import box_iou
8
  from PIL import Image
9
  import albumentations as A
 
4
  import os
5
  import torch.nn as nn
6
  import numpy as np
7
+ import torchvision
8
  from torchvision.ops import box_iou
9
  from PIL import Image
10
  import albumentations as A