File size: 959 Bytes
fd3da6c
d824c99
fd3da6c
d824c99
c59bbb9
d824c99
 
a3476e7
9379b4a
3fa06fd
d824c99
 
a3476e7
 
d824c99
 
9379b4a
d824c99
 
64ab3a9
3fa06fd
d824c99
 
a3476e7
 
d824c99
0dcfdd0
d824c99
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
import gradio as gr
from transformers import pipeline

fine_tuned_model = "Amitesh007/text_generation-finetuned-gpt2"
def generate(text):
    generator = pipeline('text-generation', model=fine_tuned_model)
    result = generator(text, num_return_sequences=2,
    max_length=100
                      
     )
    return result[0]["generated_text"],result[1]["generated_text"]

Examples = [["A light snow had fallen the night before, and there were"],["The pig face had been smashed in with a mace, but Tyrion"]]

demo = gr.Interface(
    fn=generate,
    inputs=[gr.Textbox(lines=5, label="Input Text here....")],
    outputs=[
        gr.Textbox(label="Generated Text 1"),
        gr.Textbox(label="Generated Text 2")

        ],
    title = "Text Generator GPT2 Pipeline",
    description = "this is a fine-tuned base gpt2 model inference, trained on a small 'game of thrones' dataset",
    examples = Examples
)
demo.queue(concurrency_count=2)
demo.launch()