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()
|