Spaces:

anfruizhu
/

phenotyping_pipeline

Sleeping

anfruizhu commited on Apr 21

Commit

d0e4e39

verified ·

1 Parent(s): db67212

Update bgremover.py

Files changed (1) hide show

bgremover.py CHANGED Viewed

@@ -365,8 +365,8 @@ class DamageClassifier():
                 model.load_state_dict(torch.load(model_filepath, map_location='cpu'))
             model.eval()
-            self.model = model
-            self.model_name = model_name
         return

                 model.load_state_dict(torch.load(model_filepath, map_location='cpu'))
             model.eval()
+        self.model = model
+        self.model_name = model_name
         return