drzeeIslam commited on
Commit
c227390
·
verified ·
1 Parent(s): cf3e2cb

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +12 -3
README.md CHANGED
@@ -1,10 +1,19 @@
 
 
 
 
 
 
 
 
 
1
  # Nelson-GPT Training Space
2
 
3
  This Space trains a DistilBERT model on pediatric chunks using Masked Language Modeling (MLM).
4
 
5
- - Dataset: `your-username/nelson-gpt-chunks`
6
- - Model: `distilbert-base-uncased`
7
 
8
  ## Run
9
 
10
- This Space will automatically start training when deployed.
 
1
+ ---
2
+ title: Nelson-GPT Training
3
+ emoji: ⚕️
4
+ colorFrom: blue
5
+ colorTo: green
6
+ sdk: docker
7
+ pinned: false
8
+ ---
9
+
10
  # Nelson-GPT Training Space
11
 
12
  This Space trains a DistilBERT model on pediatric chunks using Masked Language Modeling (MLM).
13
 
14
+ - **Dataset**: `drzeeIslam/nelson-gpt-chunks`
15
+ - **Model**: `distilbert-base-uncased`
16
 
17
  ## Run
18
 
19
+ This Space will automatically start training when deployed.