Spaces:
Runtime error
Runtime error
update config file
Browse files
model.py
CHANGED
|
@@ -23,7 +23,7 @@ class Model:
|
|
| 23 |
|
| 24 |
|
| 25 |
def _load_model(self, name: str) -> nn.Module:
|
| 26 |
-
return init_detector(
|
| 27 |
|
| 28 |
def set_model(self, name: str) -> None:
|
| 29 |
if name == self.model_name:
|
|
|
|
| 23 |
|
| 24 |
|
| 25 |
def _load_model(self, name: str) -> nn.Module:
|
| 26 |
+
return init_detector(config = 'https://github.com/open-mmlab/mmdetection/tree/master/configs/faster_rcnn/faster_rcnn_r50_fpn_1x_coco.py',checkpoint = 'models/orgaquant_pretrained.pth' , device=self.device)
|
| 27 |
|
| 28 |
def set_model(self, name: str) -> None:
|
| 29 |
if name == self.model_name:
|