t5-small-en-hi-v1

Model Card for Model ID

<!-- Provide a quick summary of what the model is/does. -->

Model Details

Model Description

<!-- Provide a longer summary of what this model is. -->

• Developed by: [Vishal Prem]
• Model type: [Machine Translation]
• Language(s) (NLP): [English,Hindi]
• Finetuned from model [optional]: [Flan-T5]

Uses

<!-- Address questions around how the model is intended to be used, including the foreseeable users of the model and those affected by the model. -->

Direct Use

from transformers import AutoModelForSeq2SeqLM, AutoTokenizer, T5Tokenizer, T5ForConditionalGeneration
hf_repo_name = "Vishal94/t5-small-en-hi-v1"

# Load model and tokenizer from Hugging Face Hub
tokenizer = T5Tokenizer.from_pretrained(hf_repo_name)
model = T5ForConditionalGeneration.from_pretrained(hf_repo_name)

# Example input
input_text = "Translate English to Hindi: What are you doing today?"


# Tokenize input
inputs = tokenizer(input_text, return_tensors="pt", padding=True)

# Generate output
outputs = model.generate(**inputs, max_new_tokens=64)
result = tokenizer.decode(outputs[0], skip_special_tokens=True)

print("Output:", result)