fg-mindee
commited on
Commit
•
fc4564b
1
Parent(s):
7910142
docs: Updated README
Browse files
README.md
CHANGED
|
@@ -56,7 +56,7 @@ from torchvision.transforms import Compose, ConvertImageDtype, Normalize, PILToT
|
|
| 56 |
from torchvision.transforms.functional import InterpolationMode
|
| 57 |
from holocron.models import model_from_hf_hub
|
| 58 |
|
| 59 |
-
model = model_from_hf_hub("frgfm/
|
| 60 |
|
| 61 |
img = Image.open(path_to_an_image).convert("RGB")
|
| 62 |
|
| 56 |
from torchvision.transforms.functional import InterpolationMode
|
| 57 |
from holocron.models import model_from_hf_hub
|
| 58 |
|
| 59 |
+
model = model_from_hf_hub("frgfm/cspdarknet53_mish").eval()
|
| 60 |
|
| 61 |
img = Image.open(path_to_an_image).convert("RGB")
|
| 62 |
|