Update README.md

README.md

@@ -20,6 +20,8 @@ metrics:
 ### *Predicting Clinical Neuroplasticity via Deep Genomic Regression*
 
 [![Hugging Face Spaces](https://img.shields.io/badge/%F0%9F%A4%97%20Hugging%20Face-Spaces-blue)](https://huggingface.co/spaces/assix-research/stroke-recovery-analyser)
+[![Hardware: NVIDIA DGX Spark](https://img.shields.io/badge/Hardware-NVIDIA%20DGX%20Spark-76b900)](https://www.nvidia.com/en-us/data-center/dgx-spark/)
+
 ---
 
 ## 🚀 Project Overview
@@ -27,11 +29,15 @@ This model is a specialized **Sequence-to-Scalar Transformer** designed to predi
 
 The model serves as the **Biological Modality** in a multi-modal fusion pipeline, intended to be integrated with MRI-based lesion volumetry to create personalized 3D recovery forecasts.
 
+
+
 ---
 
 ### 🎮 Live Demo
 **Test the model in real-time here:** [**Stroke-Recovery-Analyser Space**](https://huggingface.co/spaces/assix-research/stroke-recovery-analyser)
 
+---
+
 ## 🛠 Technical Specifications
 - **Base Backbone:** `InstaDeepAI/nucleotide-transformer-v2-500m-multi-species`
 - **Architecture Head:** Custom Linear Regression Head for continuous score prediction.
@@ -49,10 +55,9 @@ The model serves as the **Biological Modality** in a multi-modal fusion pipeline
 | **Optimizer** | AdamW (Weight Decay = 0.01) |
 
 ## 📊 Dataset: ARC-Aphasia
-The model was fine-tuned on a curated genomic subset of the **Aphasia Recovery Cohort (ARC)**, consisting of 902 subjects. The training focus was localized to genetic regions associated with neural growth factors and neurotransmitter regulation, including:
-- **BDNF** (Brain-Derived Neurotrophic Factor)
-- **COMT** (Catechol-O-methyltransferase)
-- **APOE** (Apolipoprotein E)
+The model was fine-tuned on a curated genomic subset of the **Aphasia Recovery Cohort (ARC)**, consisting of 902 subjects. The training focus was localized to genetic regions associated with neuroplasticity-related markers, including **BDNF**, **COMT**, and **APOE**.
+
+
 
 ## 🖥️ Usage & Inference
 To run this model locally on your DGX or a compatible Blackwell environment:
@@ -72,6 +77,7 @@ with torch.no_grad():
 
 print(f"Predicted WAB-AQ: {prediction:.2f}/100")
 ```
+
 ⚠️ Clinical Disclaimer
 
-This model is a Biomedical Research Tool developed for the study of neuroplasticity. It is not an FDA-cleared diagnostic device. Predicted WAB-AQ scores are statistical estimates based on a specific research cohort and must be interpreted by a neurologist in conjunction with structural neuroimaging (MRI/DTI).
+This model is a Biomedical Research Tool developed for the study of neuroplasticity. Predicted WAB-AQ scores are statistical estimates based on a specific research cohort and must be interpreted by a neurologist in conjunction with structural neuroimaging (MRI/DTI).