Spaces:

rajora
/

binary-classification-galaxies

Sleeping

rajora commited on Jul 19

Commit

340fe1a

•

1 Parent(s): 5713811

gpu to cpu 2

Files changed (2) hide show

utils/__pycache__/model.cpython-310.pyc CHANGED Viewed

Binary files a/utils/__pycache__/model.cpython-310.pyc and b/utils/__pycache__/model.cpython-310.pyc differ

utils/model.py CHANGED Viewed

@@ -20,6 +20,6 @@ def load_model():
     device = torch.device("cpu")
     net = EffNet(n_classes=2).to(device)
     model_path = os.path.join(os.path.dirname(__file__), 'models', 'modelo_galaxias.pth')
-    net.load_state_dict(torch.load(model_path))  # Adjust path if needed
     net.eval()
     return net

     device = torch.device("cpu")
     net = EffNet(n_classes=2).to(device)
     model_path = os.path.join(os.path.dirname(__file__), 'models', 'modelo_galaxias.pth')
+    net.load_state_dict(torch.load(model_path, map_location=torch.device('cpu')))  # Adjust path if needed
     net.eval()
     return net