Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
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.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('NailongKiller.yolo11x.v0.1.0.pt').to(device)
|
| 12 |
|
| 13 |
@spaces.GPU
|
| 14 |
def predict(img):
|