update

app.py

@@ -13,13 +13,11 @@ except:
     print('Fail to download code')
 
 try:
-    from bridge_hand_detect2.predict import CardDetectionModel, draw_bboxes
+    from bridge_hand_detect2.predict import CardDetectionModel
     from bridge_hand_detect2.pbn import create_pbn_file
-    model_dir = 'bridge_hand_detect2/model.onnx'
 except:
-    from bridge_hand_detect.predict import CardDetectionModel, draw_bboxes
+    from bridge_hand_detect.predict import CardDetectionModel
     from bridge_hand_detect.pbn import create_pbn_file
-    model_dir = 'bridge_hand_detect/model.onnx'
 
 custom_css = \
 """
@@ -32,7 +30,7 @@ custom_css = \
 INPUT_IMG_HEIGHT = 480
 OUTPUT_IMG_HEIGHT = 320 
 css = ".output_img {display:block; margin-left: auto; margin-right: auto}"
-model = CardDetectionModel(model_dir)
+model = CardDetectionModel()
 
 def predict(image_path):
     start = timer()