Matteo Sirri commited on
Commit
23918e1
·
1 Parent(s): 369cd4c

fix: fix path

Browse files
Files changed (1) hide show
  1. app.py +2 -1
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()