SamDaLamb commited on
Commit
bbb19a9
·
verified ·
1 Parent(s): 33d530c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -20,7 +20,7 @@ import matplotlib.pyplot as plt
20
  from io import BytesIO
21
 
22
  # Load the YOLO model
23
- model_path = "./best.pt"
24
  model = torch.jit.load(model_path, map_location=torch.device("cpu"))
25
  model.eval()
26
 
 
20
  from io import BytesIO
21
 
22
  # Load the YOLO model
23
+ model_path = "./last.pt"
24
  model = torch.jit.load(model_path, map_location=torch.device("cpu"))
25
  model.eval()
26