RaynWu2002 commited on
Commit
d5d388a
·
verified ·
1 Parent(s): ea70ad6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -4,7 +4,7 @@ import numpy as np
4
  import torch
5
  import os
6
  import cv2
7
- import mmcv
8
  from PIL import Image
9
  import torchvision.transforms as transforms
10
  from net.dornet import Net
@@ -21,6 +21,8 @@ def install_mmcv():
21
  print("Failed to install mmcv-full:", e)
22
 
23
  install_mmcv()
 
 
24
  print("=" * 50)
25
  print(f"CUDA available: {torch.cuda.is_available()}")
26
  print(f"torch version: {torch.__version__}")
 
4
  import torch
5
  import os
6
  import cv2
7
+
8
  from PIL import Image
9
  import torchvision.transforms as transforms
10
  from net.dornet import Net
 
21
  print("Failed to install mmcv-full:", e)
22
 
23
  install_mmcv()
24
+ import mmcv
25
+
26
  print("=" * 50)
27
  print(f"CUDA available: {torch.cuda.is_available()}")
28
  print(f"torch version: {torch.__version__}")