Hugging Face's logo

Spaces:
shvardhan
/
object_detection
Runtime error

App Files Community
shvardhan commited on
Commit
8f9ebd0
·
1 Parent(s): a4a437a

update model.py

Browse files
Files changed (1) hide show
  1. model.py +0 -1
model.py CHANGED
@@ -23,7 +23,6 @@ class Model:
23
 
24
 
25
  def _load_model(self, name: str) -> nn.Module:
26
- dic = self.MODEL_DICT[name]
27
  return init_detector('configs/_base_/faster-rcnn_r50_fpn_1x_coco.py','models/orgaquanT-pretarined.pth' , device=self.device)
28
 
29
  def set_model(self, name: str) -> None:
 
23
 
24
 
25
  def _load_model(self, name: str) -> nn.Module:
 
26
  return init_detector('configs/_base_/faster-rcnn_r50_fpn_1x_coco.py','models/orgaquanT-pretarined.pth' , device=self.device)
27
 
28
  def set_model(self, name: str) -> None: