SivaMallikarjun commited on
Commit
b49f5b6
·
verified ·
1 Parent(s): 676f3ef

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +1 -1
main.py CHANGED
@@ -4,7 +4,7 @@ from datasets import load_dataset
4
  from transformers import AutoModelForSequenceClassification, TrainingArguments, Trainer, AutoTokenizer
5
 
6
  # Load dataset
7
- dataset = load_dataset("your_dataset")
8
 
9
  # Load tokenizer and model
10
  tokenizer = AutoTokenizer.from_pretrained("distilbert-base-uncased")
 
4
  from transformers import AutoModelForSequenceClassification, TrainingArguments, Trainer, AutoTokenizer
5
 
6
  # Load dataset
7
+ dataset = load_dataset("csv", data_files={"train": "train_data.csv", "test": "test_data.csv"})
8
 
9
  # Load tokenizer and model
10
  tokenizer = AutoTokenizer.from_pretrained("distilbert-base-uncased")