File size: 737 Bytes
10d53e4
 
 
 
 
 
00202df
10d53e4
 
 
 
 
 
 
 
 
 
 
a1f52e3
cea5cac
2baa1da
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
import os
import gradio as gr
from transformers import pipeline


def blog_gen(txt):
    generator=pipeline(task='text-generation',model='shiv')
    gen_blog=generator(txt,max_length=300,num_return_sequences=2)
    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.Textbox(label="Generated Blog")],
                    css="footer {visibility: hidden}")
iface.launch()