Hugging Face's logo

Spaces:
drzeeIslam
/
Training_nelson
Runtime error

App Files Community
drzeeIslam commited on
Commit
cf3e2cb
·
verified ·
1 Parent(s): 865ef14

Update train.py

Browse files
Files changed (1) hide show
  1. train.py +1 -1
train.py CHANGED
@@ -8,7 +8,7 @@ from transformers import (
8
  )
9
 
10
  # Load dataset from Hub
11
- dataset = load_dataset("your-username/nelson-gpt-chunks")
12
 
13
  # Load tokenizer and model
14
  model_checkpoint = "distilbert-base-uncased"
 
8
  )
9
 
10
  # Load dataset from Hub
11
+ dataset = load_dataset("drzeeIslam/nelson-gpt-chunks")
12
 
13
  # Load tokenizer and model
14
  model_checkpoint = "distilbert-base-uncased"