Update app.py

app.py

@@ -10,12 +10,12 @@ from utils import non_max_suppression, plot_image, cells_to_bboxes
 import config
 import albumentations as A
 from albumentations.pytorch import ToTensorV2
-from model import YoloVersion3
+from model import YOLOv3
 import matplotlib.pyplot as plt
 import matplotlib.patches as patches
 
 # Load the model
-model = YoloVersion3( )
+model = YOLOv3( )
 model.load_state_dict(torch.load('Yolov3.pth', map_location=torch.device('cpu')), strict=False)
 model.eval()