Hugging Face's logo

Spaces:
ladyzoe
/
bear-detector-api-docker
Sleeping

App Files Community
ladyzoe commited on
Commit
c1525b2
·
verified ·
1 Parent(s): 666a9b9

best.onnx

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -14,7 +14,7 @@ app = FastAPI()
14
 
15
  # 載入 YOLOv8 模型
16
  try:
17
- model = YOLO("best.pt")
18
  print("YOLOv8 模型在 Hugging Face Space 載入成功！")
19
  except Exception as e:
20
  # 如果模型載入失敗，在啟動時就拋出錯誤
 
14
 
15
  # 載入 YOLOv8 模型
16
  try:
17
+ model = YOLO("best.onnx")
18
  print("YOLOv8 模型在 Hugging Face Space 載入成功！")
19
  except Exception as e:
20
  # 如果模型載入失敗，在啟動時就拋出錯誤