Spaces:

shvardhan
/

object_detection

Runtime error

shvardhan commited on Mar 3, 2024

Commit

d931056

1 Parent(s): 504524a

update config file

Files changed (1) hide show

model.py CHANGED Viewed

@@ -23,7 +23,7 @@ class Model:
     def _load_model(self, name: str) -> nn.Module:
-        return init_detector('configs/_base_/faster-rcnn_r50_fpn_1x_coco.py','models/orgaquant-pretrained.pth' , device=self.device)
     def set_model(self, name: str) -> None:
         if name == self.model_name:

     def _load_model(self, name: str) -> nn.Module:
+        return init_detector('configs/_base_/faster-rcnn_r50_fpn_1x_coco.py', 'models/orgaquant_pretrained.pth' , device=self.device)
     def set_model(self, name: str) -> None:
         if name == self.model_name: