ning8429 commited on
Commit
b17860b
·
verified ·
1 Parent(s): 3c526fc

Update api_server.py

Browse files
Files changed (1) hide show
  1. api_server.py +1 -0
api_server.py CHANGED
@@ -4,6 +4,7 @@ import numpy as np
4
  from PIL import Image
5
  import torchvision.transforms as transforms
6
  from pathlib import Path
 
7
 
8
  # Disable tensorflow warnings
9
  os.environ['TF_CPP_MIN_LOG_LEVEL'] = '3'
 
4
  from PIL import Image
5
  import torchvision.transforms as transforms
6
  from pathlib import Path
7
+ from ultralytics import YOLO
8
 
9
  # Disable tensorflow warnings
10
  os.environ['TF_CPP_MIN_LOG_LEVEL'] = '3'