Upload folder using huggingface_hub

README.md

@@ -1,17 +1,3 @@
----
-license: apache-2.0
-language:
-- en
-base_model:
-- nasa-impact/nasa-smd-ibm-st-v2
-tags:
-- GNN
-- NASA
-- EOSDIS
-- Earth Science
-datasets:
-- nasa-gesdisc/nasa-eo-knowledge-graph
----
 # EOSDIS Graph Neural Network Model Card
 
 ## Model Overview
@@ -23,8 +9,65 @@ datasets:
 **Base Language Model**: nasa-impact/nasa-smd-ibm-st-v2
 
 ## Model Architecture
+
 ### Architecture Diagram
+```mermaid
+graph TB
+    subgraph Input Layer
+        NASA[NASA Language Model<br/>768d] --> EMB[Text Embeddings]
+        EMB --> DS[Dataset Nodes]
+        EMB --> PUB[Publication Nodes]
+        EMB --> INST[Instrument Nodes]
+        EMB --> PLAT[Platform Nodes]
+        EMB --> KEY[Science Keyword Nodes]
+    end
+
+    subgraph GNN Layers
+        L1[GNN Layer 1<br/>768d → 256d]
+        L2[GNN Layer 2<br/>256d → 256d]
+        L3[GNN Layer 3<br/>256d → 768d]
+        
+        DS --> L1
+        PUB --> L1
+        INST --> L1
+        PLAT --> L1
+        KEY --> L1
+        
+        L1 --> L2
+        L2 --> L3
+    end
+
+    subgraph Output Layer
+        L3 --> DSO[Dataset Embeddings]
+        L3 --> PUBO[Publication Embeddings]
+        L3 --> INSTO[Instrument Embeddings]
+        L3 --> PLATO[Platform Embeddings]
+        L3 --> KEYO[Keyword Embeddings]
+    end
+
+    style NASA fill:#f9f,stroke:#333,stroke-width:2px
+    style L1 fill:#bbf,stroke:#333,stroke-width:2px
+    style L2 fill:#bbf,stroke:#333,stroke-width:2px
+    style L3 fill:#bbf,stroke:#333,stroke-width:2px
+```
 
+### Edge Types
+```mermaid
+graph LR
+    DS[Dataset] -->|references| PUB[Publication]
+    DS -->|uses| INST[Instrument]
+    DS -->|platform| PLAT[Platform]
+    DS -->|tagged_with| KEY[Science Keyword]
+    INST -->|on| PLAT
+    PUB -->|mentions| INST
+    PUB -->|discusses| KEY
+    
+    style DS fill:#f9f,stroke:#333,stroke-width:2px
+    style PUB fill:#bbf,stroke:#333,stroke-width:2px
+    style INST fill:#bfb,stroke:#333,stroke-width:2px
+    style PLAT fill:#fbb,stroke:#333,stroke-width:2px
+    style KEY fill:#ffb,stroke:#333,stroke-width:2px
+```
 
 ### Core Components
 - **Base Text Encoder**: NASA-SMD-IBM Language Model (768-dimensional embeddings)
@@ -46,10 +89,10 @@ datasets:
 - **Source**: NASA EOSDIS Knowledge Graph
 - **Node Types and Counts**:
   - Datasets: Earth science datasets from NASA DAACs
-  - Publications: Publications citing NASA EOSDIS datsets and publications citing those publications
+  - Publications: Related scientific papers
   - Instruments: Earth observation instruments
   - Platforms: Satellite and other observation platforms
-  - Science Keywords: NASA GCMD Science Keywords
+  - Science Keywords: NASA Earth Science taxonomy
 
 ### Training Process
 - **Optimization**: Adam optimizer
@@ -103,7 +146,7 @@ from gnn_model import EOSDIS_GNN
 # Load models
 tokenizer = AutoTokenizer.from_pretrained("nasa-impact/nasa-smd-ibm-st-v2")
 text_model = AutoModel.from_pretrained("nasa-impact/nasa-smd-ibm-st-v2")
-gnn_model = EOSDIS_GNN.from_pretrained("arminmehrabian/nasa-eosdis-heterogeneous-gnn")
+gnn_model = EOSDIS_GNN.from_pretrained("your-username/eosdis-gnn")
 
 # Process query
 def get_embedding(text):
@@ -167,15 +210,16 @@ results = searcher.search(
 ### Citation
 
 ```bibtex
-@misc{eosdis-gnn-2025,
+@misc{eosdis-gnn-2024,
   title={EOSDIS Graph Neural Network Model},
-  author={Armin Mehrabian},
-  year={2025},
+  author={Your Name},
+  year={2024},
   publisher={Hugging Face},
-  howpublished={\url{https://huggingface.co/arminmehrabian/nasa-eosdis-heterogeneous-gnn}}
+  howpublished={\url{https://huggingface.co/your-username/eosdis-gnn}}
 }
 ```
 
 ## Contact Information
-- **Maintainer**: [Armin Mehrabian]
-- **Email**: [[email protected]]
+- **Maintainer**: [Your Name]
+- **Email**: [Your Email]
+- **Organization**: [Your Organization] 

config.pt

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:e838a96fe36ddcd3e4d2cf8be2a6fa41c27b40cc3933194d5877f94cf8a3cc0b
-size 860
+oid sha256:b56569d9bd26838846528abb38bf37f374748b9ee790e5a37e189657824cc0de
+size 1372

pytorch_model.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:7fb26177ceb863df2c3b1fc94c0ecf691cff19abcc515c163f8eee2cba81e080
+oid sha256:04a0203feb172b10c369032acfa1cf46d419bf0ba744dfcf9f7644e67f0191bb
 size 27120886