Spaces:

Megatron17
/

Lightning-YOLO

Runtime error

Megatron17 commited on Aug 18, 2023

Commit

f9c5d15

1 Parent(s): bb78a4a

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -7,7 +7,7 @@ import torch
 scaled_anchors = config.scaled_anchors
 model = Lightning_YOLO()
-model.load_state_dict(torch.load("model.ckpt", map_location="cpu"), strict=False)
 model.eval()
 def inference(image, threst = 0.5, iou_tresh = 0.5):

 scaled_anchors = config.scaled_anchors
 model = Lightning_YOLO()
+model.load_state_dict(torch.load("yolov3.pth", map_location="cpu"), strict=False)
 model.eval()
 def inference(image, threst = 0.5, iou_tresh = 0.5):