Spaces:

cvachet
/

object_detection_gradio

Running

Clement Vachet commited on Jan 6

Commit

5eda9a8

1 Parent(s): ae565a0

Fix: fix model eval

Files changed (1) hide show

detect_torch.py CHANGED Viewed

@@ -95,7 +95,7 @@ def detect(im, model, transform):
 def load_model():
     model = torch.hub.load('facebookresearch/detr', 'detr_resnet50', pretrained=True)
-    model.eval();
     return model
 def main():

 def load_model():
     model = torch.hub.load('facebookresearch/detr', 'detr_resnet50', pretrained=True)
+    model.eval()
     return model
 def main():