Upload folder using huggingface_hub

MLBaseModel.py

@@ -14,4 +14,4 @@ class MLBaseModel(nn.Module):
         x = torch.relu(self.fc2(x))
         x = torch.relu(self.fc3(x))
         x = self.fc4(x)
-        return x
+        return x

MLBaseModelDriver.py

@@ -61,7 +61,7 @@ class MLBaseModelDriver:
         model_class = self._import_model_class(model_class_file)
 
         model = model_class(input_dim=4)
-        state_dict = torch.load(model_file, map_location=torch.device("cpu"), weights_only=False)
+        state_dict = torch.load(model_file, weights_only=False)
         model.load_state_dict(state_dict)
         model.eval()
 
@@ -80,7 +80,7 @@ class MLBaseModelDriver:
         download files from hugging face
         :return: downloaded files
         """
-        model_path = "ckoozzzu/NextPlace"
+        model_path = "Nickel5HF/NextPlace"
 
         # Download the model files from the Hugging Face Hub
         model_file = hf_hub_download(repo_id=model_path, filename="model_files/real_estate_model.pth")

model_files/real_estate_model.pth

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:9ead0cc482ac184bf4aff9500dcfd288304a243200621b09574713fb20fa6dbe
-size 2079339
+oid sha256:c67cd8a4b692510c9e21889f6a5d0fcae0d5a9d7995c0629cdc3fa08fbb2c61f
+size 2152506