import gradio as gr
from inference import predict

def compare(goal, sol1, sol2):
    return predict(goal, sol1, sol2)

demo = gr.Interface(
    fn=compare,
    inputs=[gr.Text(label="Goal"), gr.Text(label="Solution 1"), gr.Text(label="Solution 2")],
    outputs=gr.Text(label="Preferred Solution"),
    title="EvoTransformer v2.1 - PIQA Commonsense Reasoning"
)

demo.launch()