From 42469f357248527a8dedd9c46eb21a1f22e786db Mon Sep 17 00:00:00 2001 From: Piotr Szkudlarek Date: Mon, 8 Jan 2024 14:05:02 +0100 Subject: [PATCH] Change model weights location --- PlanktonDetector/DetectionApp/utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/PlanktonDetector/DetectionApp/utils.py b/PlanktonDetector/DetectionApp/utils.py index 7eb3789..5829125 100644 --- a/PlanktonDetector/DetectionApp/utils.py +++ b/PlanktonDetector/DetectionApp/utils.py @@ -5,7 +5,7 @@ from io import BytesIO from django.conf import settings from ultralytics import YOLO -MODEL = YOLO("best.pt") +MODEL = YOLO("../best.pt") def predict_image(image):