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
|