Update README.md

README.md

@@ -32,58 +32,46 @@ This is a Vision Transformer (ViT) model fine-tuned using Low-Rank Adaptation (L
 2. Start frequency (Hz)
 3. End frequency (Hz)
 
-### Architecture
-
-- **Base Model**: Pre-trained Vision Transformer (ViT)
-- **Adaptation Method**: LoRA (Low-Rank Adaptation)
-- **Framework**: PyTorch
-- **Task**: Regression on time-frequency representations
+<div style="background: #f8f9fa; border-radius: 8px; padding: 20px; margin-bottom: 20px; border-left: 4px solid #4285f4;">
+<h2 style="margin-top: 0;">🔧 Fine-Tuning Details</h2>
+<div style="background: white; border-radius: 8px; padding: 15px; box-shadow: 0 2px 4px rgba(0,0,0,0.1);">
+    <ul>
+        <li><strong>Framework:</strong> PyTorch</li>
+        <li><strong>Architecture:</strong> Pre-trained Vision Transformer (ViT)</li>
+        <li><strong>Adaptation Method:</strong> LoRA (Low-Rank Adaptation)</li>
+        <li><strong>Task:</strong> Regression on time-frequency representations</li>
+        <li><strong>Training Protocol:</strong> Automatic Mixed Precision (AMP), Early stopping, Learning Rate scheduling</li>
+        <li><strong>Output:</strong> Quantitative predictions + optional natural language descriptions</li>
+    </ul>
+</div>
+</div>
 
-<div style="background: #f8f9fa; border-radius: 8px; padding: 20px; margin-bottom: 20px; border-left: 4px solid #ea4335;">
-<h2 style="margin-top: 0;">🔗 Dataset Sources</h2>
+<div style="background: #f8f9fa; border-radius: 8px; padding: 20px; margin-bottom: 20px; border-left: 4px solid #34a853;">
+<h2 style="margin-top: 0;">📦 Resources</h2>
 <div style="display: flex; flex-wrap: wrap; gap: 15px;">
     <div style="flex: 1; min-width: 250px; background: white; border-radius: 8px; padding: 15px; box-shadow: 0 2px 4px rgba(0,0,0,0.1);">
-        <h4 style="margin-top: 0;">GitHub Repository</h4>
-        <p><a href="https://github.com/nbahador/chirp_spectrogram_generator/tree/main">Chirp Spectrogram Generator</a></p>
+        <h4 style="margin-top: 0;">Trained Model</h4>
+        <p><a href="https://huggingface.co/nubahador/Fine_Tuned_Transformer_Model_for_Chirp_Localization/tree/main">HuggingFace Model Hub</a></p>
     </div>
     <div style="flex: 1; min-width: 250px; background: white; border-radius: 8px; padding: 15px; box-shadow: 0 2px 4px rgba(0,0,0,0.1);">
-        <h4 style="margin-top: 0;">Research Paper</h4>
-        <p><a href="https://arxiv.org/pdf/2503.22713">ArXiv Publication</a></p>
+        <h4 style="margin-top: 0;">Spectrogram Dataset</h4>
+        <p><a href="https://huggingface.co/datasets/nubahador/ChirpLoc100K___A_Synthetic_Spectrogram_Dataset_for_Chirp_Localization/tree/main">HuggingFace Dataset Hub</a></p>
+    </div>
+    <div style="flex: 1; min-width: 250px; background: white; border-radius: 8px; padding: 15px; box-shadow: 0 2px 4px rgba(0,0,0,0.1);">
+        <h4 style="margin-top: 0;">PyTorch Implementation</h4>
+        <p><a href="https://github.com/nbahador/Train_Spectrogram_Transformer">GitHub Repository</a></p>
+    </div>
+    <div style="flex: 1; min-width: 250px; background: white; border-radius: 8px; padding: 15px; box-shadow: 0 2px 4px rgba(0,0,0,0.1);">
+        <h4 style="margin-top: 0;">Chirp Generator</h4>
+        <p><a href="https://github.com/nbahador/chirp_spectrogram_generator">GitHub Package</a></p>
     </div>
 </div>
 </div>
 
-<div style="background: #f8f9fa; border-radius: 8px; padding: 20px; margin-bottom: 20px; border-left: 4px solid #fbbc05;">
-<h2 style="margin-top: 0;">🗂 Dataset Structure</h2>
-<div style="background: white; border-radius: 8px; padding: 15px; box-shadow: 0 2px 4px rgba(0,0,0,0.1); margin-bottom: 15px;">
-    <h4 style="margin-top: 0;">Content Includes:</h4>
-    <ul>
-        <li>Spectrogram images</li>
-        <li>Corresponding labels including chirp parameters and locations</li>
-        <li>Metadata about generation parameters</li>
-    </ul>
-</div>
-
-<div style="background: white; border-radius: 8px; padding: 15px; box-shadow: 0 2px 4px rgba(0,0,0,0.1);">
-    <h4 style="margin-top: 0;">Data Collection and Processing</h4>
-    <p>Data was synthetically generated using the provided Python package, with parameters randomly sampled from physiologically relevant ranges.</p>
-    <p><strong>Repository:</strong> <a href="https://github.com/nbahador/chirp_spectrogram_generator/tree/main">GitHub Package</a></p>
-</div>
-</div>
-
-<div style="background: #f8f9fa; border-radius: 8px; padding: 20px; margin-bottom: 20px; border-left: 4px solid #673ab7;">
+<div style="background: #f8f9fa; border-radius: 8px; padding: 20px; border-left: 4px solid #ea4335;">
 <h2 style="margin-top: 0;">📄 Citation</h2>
 <div style="background: white; border-radius: 8px; padding: 15px; box-shadow: 0 2px 4px rgba(0,0,0,0.1);">
+    <p>If you use this model in your research, please cite:</p>
     <p>Bahador, N., & Lankarany, M. (2025). Chirp localization via fine-tuned transformer model: A proof-of-concept study. arXiv preprint arXiv:2503.22713. <a href="https://arxiv.org/pdf/2503.22713">[PDF]</a></p>
 </div>
-</div>
-
-<div style="background: #f8f9fa; border-radius: 8px; padding: 20px; border-left: 4px solid #00bcd4;">
-<h2 style="margin-top: 0;">ℹ️ More Information</h2>
-<p>For more information and generation code, visit the <a href="https://github.com/nbahador/Train_Spectrogram_Transformer">GitHub repository</a>.</p>
-
-<div style="margin-top: 15px; padding-top: 15px; border-top: 1px solid #e0e0e0;">
-    <h4 style="margin-bottom: 5px;">Card Author</h4>
-    <p><a href="https://www.linkedin.com/in/nooshin-bahador-30348950/">Nooshin Bahador</a></p>
-</div>
 </div>