Fix: update app and requirement

app.py

@@ -47,7 +47,6 @@ class Pytorch_model:
         :param min_area: 小于该尺度的bbox忽略
         :return:
         '''
-        img = cv2.imread(img)
         if self.img_channel == 3:
             img = cv2.cvtColor(img, cv2.COLOR_BGR2RGB)
         h, w = img.shape[:2]
@@ -59,10 +58,10 @@ class Pytorch_model:
 
         tensor = tensor.to(self.device)
         with torch.no_grad():
-            torch.cuda.synchronize(self.device)
+            # torch.cuda.synchronize(self.device)
             start = time.time()
             preds = self.net(tensor)[0]
-            torch.cuda.synchronize(self.device)
+            # torch.cuda.synchronize(self.device)
             scale = (preds.shape[2] / w, preds.shape[1] / h)
             t = time.time() - start
             
@@ -112,7 +111,7 @@ class Pytorch_model:
 
 def run(img):
     # os.environ["CUDA_VISIBLE_DEVICES"] = "5"
-    model_path = 'pre_trained/MobileNetv2_best_loss.pth'    
+    model_path = 'MobileNetv2_best_loss.pth'    
     
     # init model 
     model = Pytorch_model(model_path, gpu_id = None)  ## set GPU id or None if you only have cpu

requirements.txt

@@ -5,3 +5,4 @@ numpy
 pyclipper
 gradio
 matplotlib
+shapely