Added model eval() mode

app.py

@@ -13,6 +13,8 @@ model = get_model('efficientDet_icevision.ckpt')
 print('Loading the classifier')
 classifier = CustomViT(target_size=7, pretrained=False)
 classifier.load_state_dict(torch.load('class_ViT_taco_7_class.pth', map_location='cpu'))
+# Set eval mode to deactivate dropout and BN layers 
+classifier.eval()
 
 def plot_img_no_mask(image, boxes, labels):
     colors = {