Hugging Face's logo

Spaces:
Hakureirm
/
NailongKiller
Sleeping

App Files Community
Hakureirm commited on
Commit
04327f7
·
verified ·
1 Parent(s): 8e5143e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -8,7 +8,7 @@ import numpy as np
8
 
9
  app = FastAPI()
10
  device = 'cuda' if torch.cuda.is_available() else 'cpu'
11
- model = YOLO('NailongKiller.yolo11x.v0.1.0.pt').to(device)
12
 
13
  @spaces.GPU
14
  def predict(img):
 
8
 
9
  app = FastAPI()
10
  device = 'cuda' if torch.cuda.is_available() else 'cpu'
11
+ model = YOLO('nailong_yolo11.pt').to(device)
12
 
13
  @spaces.GPU
14
  def predict(img):