add albumentation library

app.py

@@ -15,7 +15,8 @@ except:
 try:
     from bridge_hand_detect2.predict import CardDetectionModel
     from bridge_hand_detect2.pbn import create_pbn_file
-except:
+except Exception as e:
+    print(e)
     from bridge_hand_detect.predict import CardDetectionModel
     from bridge_hand_detect.pbn import create_pbn_file
 
@@ -36,7 +37,7 @@ def predict(image_path):
     start = timer()
     df = None
     try:
-        hands, (width,height) = model(image_path)
+        hands, (width,height) = model(image_path, augment=True)
         print(hands)
         # Output dataframe
         df = pd.DataFrame(['♠', '♥', '♦', '♣'], columns=[''])

requirements.txt

@@ -3,4 +3,5 @@ onnx
 onnxruntime
 lap
 opencv-python
-openvino==2024.0.0
+openvino==2024.0.0
+albumentations