Hugging Face
Models
Datasets
Spaces
Community
Docs
Enterprise
Pricing
Log In
Sign Up
Spaces:
Dirake
/
Pokemon2
like
0
Runtime error
App
Files
Files
Community
Fetching metadata from the HF Docker repository...
main
Pokemon2
/
predict.py
Dirake
Upload 13 files
acbc0af
verified
4 months ago
raw
Copy download link
history
blame
contribute
delete
Safe
195 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
)