Spaces:

rachman
/

object_detection

Sleeping

rachman commited on May 22, 2025

Commit

e02cf0e

verified ·

1 Parent(s): c2d026c

Update src/prediction.py

Files changed (1) hide show

src/prediction.py CHANGED Viewed

@@ -6,7 +6,7 @@ import os
 import matplotlib.pyplot as plt
 import matplotlib.patches as patches
-model = YOLO("best.pt")
 label_map = {
     0: "scissors",

 import matplotlib.pyplot as plt
 import matplotlib.patches as patches
+model = YOLO("src/best.pt")
 label_map = {
     0: "scissors",