Spaces:
Build error
Build error
File size: 697 Bytes
c92b89d 9a9a4a3 c92b89d 25c3227 692f1c8 0a26a6b 25c3227 c92b89d 9a9a4a3 25c3227 9a9a4a3 c92b89d 9a9a4a3 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 |
import gradio as gr
from transformers import pipeline
def convert(prompt):
model = pipeline('text-generation', model='bigscience/bloom-1b1')
result = model(prompt, repetition_penalty = 2.0, max_length=200, num_beams = 2, num_beam_groups = 2, top_k=1, temperature=0.9, diversity_penalty = 0.9)
return result[0]['generated_text']
with gr.Blocks() as bls:
gr.Markdown("Here is a Text Generation app")
with gr.Row():
inp = gr.Textbox(label="Type your prompt here and click Run", placeholder='Example: The main cloud services of AWS are:')
out = gr.Textbox(label="Output")
btn = gr.Button("Run")
btn.click(fn=convert, inputs=inp, outputs=out)
bls.launch() |