Update app.py

app.py

@@ -3,13 +3,17 @@ import torch.nn as nn
 from parseq.system import System
 import yaml
 import cv2
-from parseq.augmentation import trans
 from PIL import Image
 from wpodnet.lib_detection import load_model_wpod, detect_lp
 import numpy as np
 import gradio as gr 
+from torchvision import transforms as T
 
-
+trans = T.Compose([
+            T.Resize((224, 224), T.InterpolationMode.BICUBIC),
+            T.ToTensor(),
+            T.Normalize(0.5, 0.5)
+        ])
 
 device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')