import gradio as gr
import os

gr.load("models/jpronchick/CLAUS_base_v1", hf_token=os.environ["HF_TOKEN"]).launch(server_port=8000)

def generate_output(prompt, response):
    return "A surprising use for a {} is a {}".format(prompt, response)

input_components = [
    gr.inputs.Textbox(label="Prompt"),
    gr.inputs.Textbox(label="Response")
]

output_component = gr.outputs.Textbox(label="Output")

gr.Interface(fn=generate_output, inputs=input_components, outputs=output_component, examples=["A surprising use for a shoe is a house for a worm"]).launch(server_port=8000)