Spaces:

sergio-sanz-rodriguez
/

transform-eats

Running

sergio-sanz-rodriguez commited on Dec 24, 2024

Commit

cd79e1e

1 Parent(s): 8df1ddb

bug fix

Files changed (1) hide show

model.py CHANGED Viewed

@@ -120,6 +120,6 @@ def create_effnetb0(
     model_path = model_dir_path / model_weights_name
     # Load the state dictionary into the model
-    effnetb0_model.load_state_dict(torch.load(model_path, weights_only=True, map_location=torch.device('cpu'))))
     return effnetb0_model

     model_path = model_dir_path / model_weights_name
     # Load the state dictionary into the model
+    effnetb0_model.load_state_dict(torch.load(model_path, weights_only=True, map_location=torch.device('cpu')))
     return effnetb0_model