Spaces:
Build error
Build error
Matteo Sirri
commited on
Commit
·
23918e1
1
Parent(s):
369cd4c
fix: fix path
Browse files
app.py
CHANGED
@@ -1,4 +1,5 @@
|
|
1 |
import os.path as osp
|
|
|
2 |
import gradio as gr
|
3 |
import torch
|
4 |
import logging
|
@@ -21,7 +22,7 @@ def load_model(baseline: bool = False):
|
|
21 |
in_features = model.roi_heads.box_predictor.cls_score.in_features
|
22 |
model.roi_heads.box_predictor = FastRCNNPredictor(in_features, 2)
|
23 |
checkpoint = torch.load(
|
24 |
-
"model_split_3_FT_MOT17.pth", map_location="cpu")
|
25 |
model.load_state_dict(checkpoint["model"])
|
26 |
model.to(device)
|
27 |
model.eval()
|
|
|
1 |
import os.path as osp
|
2 |
+
import os
|
3 |
import gradio as gr
|
4 |
import torch
|
5 |
import logging
|
|
|
22 |
in_features = model.roi_heads.box_predictor.cls_score.in_features
|
23 |
model.roi_heads.box_predictor = FastRCNNPredictor(in_features, 2)
|
24 |
checkpoint = torch.load(
|
25 |
+
osp.join(os.getcwd(),"model_split_3_FT_MOT17.pth"), map_location="cpu")
|
26 |
model.load_state_dict(checkpoint["model"])
|
27 |
model.to(device)
|
28 |
model.eval()
|