Upload folder using huggingface_hub

config.json

@@ -13,9 +13,9 @@
   "input_channels": 4,
   "img_embedding_dim": 512,
   "pot_features": {
-    "num_entities": 2,
+    "num_entities": 1,
     "feature_dim": 3,
-    "flattened_dim": 6,
+    "flattened_dim": 3,
     "mlp_hidden_dim": 64,
     "embedding_dim": 64
   },

pytorch_model.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:006ae7e0e5bda3e7ba6bef1e973ae553d3d0dbff1badcf8a43b02c32559704ea
-size 45204219
+oid sha256:dd64ed963cb48b0eb51ad4081aed851113b8cb53e857199cd8413f8cfe15cefe
+size 45203451