Update app.py

app.py

@@ -3,7 +3,7 @@ import gradio as gr
 import numpy as np
 from PIL import Image
 import torchvision.transforms as transforms
-from model import Yolov1
+from model import Yolo_V1
 from utils import cellboxes_to_boxes, non_max_suppression
 import cv2
 import os
@@ -35,7 +35,7 @@ except Exception as e:
     print("Tentative de chargement local...")
     model_path = MODEL_FILENAME
 
-model = Yolov1(split_size=7, num_boxes=2, num_classes=20).to(DEVICE)
+model = Yolo_V1(split_size=7, num_boxes=2, num_classes=20).to(DEVICE)
 checkpoint = torch.load(model_path, map_location=DEVICE)
 model.load_state_dict(checkpoint["state_dict"])
 model.eval()