jafrilalam's picture
Update app.py
d3a1f2f verified
raw
history blame contribute delete
952 Bytes
import gradio as gr
from transformers import pipeline, AutoModelForSeq2SeqLM, MBart50Tokenizer, AutoTokenizer
tokenizer = AutoTokenizer.from_pretrained('jafrilalam/bangla_sentence_correction_01', src_lang="bn_IN", tgt_lang="bn_IN", use_fast=True)
model = AutoModelForSeq2SeqLM.from_pretrained("jafrilalam/bangla_sentence_correction_01", use_safetensors=True)
def correct_text(given_sentence):
inputs = tokenizer.encode(
given_sentence,
truncation=True,
return_tensors="pt",
max_length=len(given_sentence),
)
output_ids = model.generate(
inputs,
max_new_tokens=len(given_sentence),
early_stopping=True,
)
return tokenizer.decode(output_ids[0], skip_special_tokens=True)
iface = gr.Interface(
fn=correct_text,
inputs=gr.Textbox(lines=4, label="Incorrect Bangla Sentence"),
outputs=gr.Textbox(label="Corrected Bangla Sentence")
)
iface.launch(share = True)