YasaswiniCh commited on
Commit
9a9c572
·
verified ·
1 Parent(s): 76a3f8d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -1,8 +1,9 @@
1
  import streamlit as st
2
  from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
 
3
 
4
- # Load your trained model and tokenizer from Hugging Face
5
- model_name = "ai4bharat/IndicTrans"
6
  tokenizer = AutoTokenizer.from_pretrained(model_name)
7
  model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
8
 
 
1
  import streamlit as st
2
  from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
3
+ from datasets import load_dataset
4
 
5
+ # Login using e.g. `huggingface-cli login` to access this dataset
6
+ ds = load_dataset("pavanteja15/english-to-telugu-570K")
7
  tokenizer = AutoTokenizer.from_pretrained(model_name)
8
  model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
9