Spaces:
Runtime error
Runtime error
Add weight files in correct directory
Browse files- app.py +1 -1
- weights/ear_YOLOv5_n.pt +3 -0
- weights/weights.pt +3 -0
app.py
CHANGED
|
@@ -19,7 +19,7 @@ ear_detector.to(device)
|
|
| 19 |
model = timm.create_model("hf-hub:BVRA/MegaDescriptor-T-224", pretrained=True, num_classes=0)
|
| 20 |
|
| 21 |
# load state dict containing miscellaneous state or just the model weights
|
| 22 |
-
state_dict = torch.load("weights.pt", map_location=device)
|
| 23 |
if "optimizer" in state_dict:
|
| 24 |
model.load_state_dict(state_dict["model"])
|
| 25 |
else:
|
|
|
|
| 19 |
model = timm.create_model("hf-hub:BVRA/MegaDescriptor-T-224", pretrained=True, num_classes=0)
|
| 20 |
|
| 21 |
# load state dict containing miscellaneous state or just the model weights
|
| 22 |
+
state_dict = torch.load("weights/weights.pt", map_location=device)
|
| 23 |
if "optimizer" in state_dict:
|
| 24 |
model.load_state_dict(state_dict["model"])
|
| 25 |
else:
|
weights/ear_YOLOv5_n.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:e7aabe974a13c15ea2492a7f1583ba13255163305d0fc9c0aff8b8ee33da6651
|
| 3 |
+
size 3923125
|
weights/weights.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:da82dcf2c50dec71cafefd803cc8d078cdcf9226cbf571b7adde0dd4b14c6e7a
|
| 3 |
+
size 224710970
|