Update app.py

app.py

@@ -1,22 +1,22 @@
 import os
 import gradio as gr
 # from app_util import ContextDetDemo
-# import torch
-# import cv2
-# import matplotlib.pyplot as plt
-# import torchvision.transforms as transforms
+import torch
+import cv2
+import matplotlib.pyplot as plt
+import torchvision.transforms as transforms
 from utils.my_model import MyCNN
 from models.common import DetectMultiBackend
-# import numpy as np
-# import csv
-# import torch.nn.functional as F
-# from PIL import Image, ImageOps
+import numpy as np
+import csv
+import torch.nn.functional as F
+from PIL import Image, ImageOps
 from utils.augmentations import letterbox
 from utils.general import (scale_boxes, non_max_suppression)
-# import pandas as pd
-# import os
+import pandas as pd
+import os
 
-# from torchvision.ops import roi_align
+from torchvision.ops import roi_align
 from utils.general import (LOGGER, Profile, check_file, check_img_size, check_imshow, check_requirements, colorstr, cv2,
                            increment_path, non_max_suppression, print_args, scale_boxes, strip_optimizer, xyxy2xywh,get_fixed_xyxy)
 # Initialize Model with Error Handling