import os
import gradio as gr
from transformers import pipeline


def blog_gen(txt):
    generator=pipeline(task='text-generation',model='shivanshu292001/GeneratorModel')
    gen_blog=generator(txt,max_length=2000)
    remove_gen_text = list(map(lambda x: x["generated_text"], gen_blog))
    clean_text = list(map(lambda x: x.replace("\n\n", " "), remove_gen_text))
    return clean_text


iface = gr.Interface(fn=blog_gen,
                     inputs=[
                         gr.inputs.Textbox(
                             lines=2, placeholder=None, label='Sentence'),
                     ],
                     outputs=[gr.outputs.JSON(label=None)],
                    css="footer {visibility: hidden}")
iface.launch()