Spaces:
Runtime error
Runtime error
import gradio as gr | |
import pip | |
pip.main(['install', 'transformers']) | |
pip.main(['install', 'torch']) | |
from transformers import pipeline | |
model_name_or_path = "dgnk007/crow" | |
def generate_response(message): | |
prompt_template=f"### instruct: {message} \n ### Bot:" | |
generate=pipeline('text-generation',model=model_name_or_path) | |
response=generate(prompt_template,max_length=1024,return_full_text=False,num_return_sequences=2) | |
return response[0]['generated_text'], response[1]['generated_text'] | |
with gr.Blocks(theme=gr.themes.Soft()) as block: | |
with gr.Row(): | |
with gr.Column(): | |
input_box = gr.Textbox(label="Enter your text here:", min_width=600) | |
prompt = gr.Button("Submit", variant="primary") | |
with gr.Row(): | |
gr.Examples(['what is google?','what is youtube?'], input_box,) | |
with gr.Row(): | |
output_block = [ | |
gr.Textbox(label="Response 1", interactive=False), | |
gr.Textbox(label="Response 2", interactive=False) | |
] | |
with gr.Row(): | |
gr.Button(value="Tie") | |
gr.Button(value="Response 1 Wins",variant='primary') | |
gr.Button(value="Response 2 Wins",variant='primary') | |
gr.Button(value="Both are Bad",variant='secondary') | |
prompt.click(fn=generate_response, inputs=input_box, outputs=output_block) | |
block.launch() |