Spaces:

aje6
/

ASL-Final-Project-ONNX

Sleeping

aje6 commited on Dec 10, 2024

Commit

ceba677

verified ·

1 Parent(s): 9d7795b

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -189,7 +189,8 @@ import torch
 import cv2
 # Load the YOLOv8 model
-model = torch.hub.load('ultralytics/yolov8', 'Model_IV.pt', pretrained=True)
 def inference(img):
     results = model(img)

 import cv2
 # Load the YOLOv8 model
+model = torch.hub.load('ultralytics/yolov8', 'yolov8s')
+model.load_state_dict(torch.load('Model_IV'))
 def inference(img):
     results = model(img)