import gradio as gr
from confidence import run_nli


def greet(query):
    results = run_nli(query, sample_size=5)
    return results


sample_list = [
    "Tell me something about Albert Einstein, e.g., birth date and place and short bio ",
]

iface = gr.Interface(fn=greet, inputs="text", outputs="text", examples=sample_list, cache_examples=True)
iface.launch()