import gradio as gr
import random



pipeline = lambda x: random.randint(0, 1)

def predict(protein_sequence: str) -> dict[str, float]:
    predictions = dict(soluble=0.42, insoluble=0.58)
    return predictions

gradio_app = gr.Interface(
    predict,
    inputs="text",
    outputs="label",
    title="Soluble or insoluble, that is the question",
)

if __name__ == "__main__":
    gradio_app.launch()