Hugging Face
Models
Datasets
Spaces
Community
Docs
Enterprise
Pricing
Log In
Sign Up
Spaces:
Dirake
/
Pokemon
like
0
Configuration error
App
Files
Files
Community
Fetching metadata from the HF Docker repository...
main
Pokemon
/
predict.py
Dirake
Create predict.py
cb8ea16
verified
2 months ago
raw
Copy download link
history
blame
contribute
delete
Safe
194 Bytes
import
ultralytics
from
ultralytics
import
YOLO
ultralytics.checks()
import
torch
device =
'cuda'
model = YOLO(
"best.pt"
)
path =
"test.mp4"
results = model.predict(source=path, show =
True
)