Fixing bugs

requirements.txt

@@ -2,5 +2,4 @@ icevision[all]
 matplotlib
 effdet
 gradio
-streamlit==1.2.0
 Pillow==8.4.0

utils.py

@@ -4,10 +4,10 @@ import numpy as np
 import cv2
 import torch
 
-from  classifier import CustomViT
+from classifier import CustomViT
 from model import get_model
 
-def plot_img_no_mask(image : np.ndarray, boxes, labels):
+def plot_img_no_mask(image : np.ndarray, boxes : torch.Tensor, labels):
     colors = {
         0: (255,255,0),
         1: (255, 0, 0),