Spaces:
Sleeping
Sleeping
import gradio as gr | |
import pandas as pd | |
import random | |
df = pd.read_csv('data.csv') | |
models = df.columns.tolist() | |
print(models) | |
models.remove('hash') | |
models.remove('message') | |
messages = sorted(df['message'].tolist(), key=len) | |
messages_select = [(m[:150],m) for m in messages] | |
def out(message, model1, model2): | |
row = df[df['message'] == message] | |
output1 = row[model1].values[0] | |
output2 = row[model2].values[0] | |
return message, output1, output2 | |
with gr.Blocks() as iface: | |
with gr.Row(): | |
drop_message = gr.Dropdown(messages_select, label='Prompt', value=random.choice(messages)) | |
with gr.Row(): | |
drop_model1 = gr.Dropdown(models, label='Model 1', value=random.choice(models)) | |
drop_model2 = gr.Dropdown(models, label='Model 2', value=random.choice(models)) | |
with gr.Row(): | |
btn = gr.Button("Run") | |
with gr.Row(): | |
out_message = gr.TextArea(label='Prompt') | |
with gr.Row(): | |
out_model1 = gr.TextArea(label='Output Model 1') | |
out_model2 = gr.TextArea(label='Output Model 2') | |
btn.click(out, | |
inputs=[drop_message, drop_model1, drop_model2], | |
outputs=[out_message, out_model1, out_model2]) | |
iface.launch() |