app.py

import gradio as gr

from transformers import AutoModelForSeq2SeqLM, AutoTokenizer
from normalizer import normalize # pip install git+https://github.com/csebuetnlp/normalizer

def summarize(text):
    model = AutoModelForSeq2SeqLM.from_pretrained("csebuetnlp/banglat5")
    tokenizer = AutoTokenizer.from_pretrained("csebuetnlp/banglat5", use_fast=False)

    input_ids = tokenizer(normalize(text), return_tensors="pt").input_ids
    generated_tokens = model.generate(input_ids)
    decoded_tokens = tokenizer.batch_decode(generated_tokens)[0]
    return decoded_tokens


gr_interface = gr.Interface("models/Afsara/fb_bart_large_cnn")

gr_interface.launch()