Spaces:
Sleeping
Sleeping
File size: 493 Bytes
b811e01 a72756f b811e01 a72756f b811e01 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 |
from ultralytics import solutions
import os
import torch
import streamlit
torch.classes.__path__ = [os.path.join(torch.__path__[0], torch.classes.__file__)]
# or simply:
torch.classes.__path__ = []
inf = solutions.Inference(
model="./model/best_100.pt", # You can use any model that Ultralytics support, i.e. YOLO11, or custom trained model
#for desktop apps
# model="best_100.pt"
)
inf.inference()
### Make sure to run the file using command `streamlit run <file-name.py>` |