rvv-karma's picture
Update README.md
90a1550
|
raw
history blame
1.33 kB
metadata
language:
  - en
  - hi
  - multilingual
tags:
  - text2text-generation
widget:
  - text: Is it snowing in New York?
    example_title: Is it snowing in New York?
  - text: Is it raining in Hawaii?
    example_title: Is it raining in Hawaii?
  - text: Turn off music.
    example_title: Turn off music.
datasets:
  - rvv-karma/English-Hinglish-TOP
license: apache-2.0
pipeline_tag: text2text-generation

English2Hinglish-Flan-T5-Base

This is a finetuned model of Flan T5 Base with English-Hinglish-TOP dataset.

Usage

# Load model directly
from transformers import AutoTokenizer, AutoModelForSeq2SeqLM

model_name = "rvv-karma/English2Hinglish-Flan-T5-Base"
tokenizer = AutoTokenizer.from_pretrained(model_name)
model = AutoModelForSeq2SeqLM.from_pretrained(model_name)

input_text = "What are you doing?"
input_ids = tokenizer(input_text, return_tensors="pt")
output_ids = model.generate(**input_ids)
output_text = tokenizer.decode(output_ids[0], skip_special_tokens=True)

print(output_text)

Fine-tuning script

Google Colaboratory Notebook

References

DataCamp