Redmind commited on
Commit
5bcc7b1
·
verified ·
1 Parent(s): 3d5b8ef

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -2,7 +2,7 @@ from datasets import load_dataset
2
  from transformers import MarianMTModel, MarianTokenizer, TrainingArguments, Trainer, DataCollatorForSeq2Seq
3
 
4
  # Load dataset
5
- dataset = load_dataset('csv', data_files='dataset.tsv', delimiter='\t')
6
 
7
  # Load MarianMT tokenizer for translation task
8
  tokenizer = MarianTokenizer.from_pretrained('Helsinki-NLP/opus-mt-en-hi')
 
2
  from transformers import MarianMTModel, MarianTokenizer, TrainingArguments, Trainer, DataCollatorForSeq2Seq
3
 
4
  # Load dataset
5
+ dataset = load_dataset('csv', data_files='hindi_dataset.tsv', delimiter='\t')
6
 
7
  # Load MarianMT tokenizer for translation task
8
  tokenizer = MarianTokenizer.from_pretrained('Helsinki-NLP/opus-mt-en-hi')