Spaces:
Running
Running
Update model_yolov5.py
Browse files- model_yolov5.py +0 -1
model_yolov5.py
CHANGED
|
@@ -5,7 +5,6 @@ from yolov5.utils.plots import Annotator, colors
|
|
| 5 |
from huggingface_hub import get_token
|
| 6 |
|
| 7 |
|
| 8 |
-
@spaces.GPU
|
| 9 |
def load_model(model_path, img_size=640):
|
| 10 |
"""Load model from HuggingFace Hub."""
|
| 11 |
model = yolov5.load(model_path, hf_token=get_token())
|
|
|
|
| 5 |
from huggingface_hub import get_token
|
| 6 |
|
| 7 |
|
|
|
|
| 8 |
def load_model(model_path, img_size=640):
|
| 9 |
"""Load model from HuggingFace Hub."""
|
| 10 |
model = yolov5.load(model_path, hf_token=get_token())
|