import gradio as gr

from transformers import pipeline

pipe = pipeline(model="fibleep/mistral-7b-mickiewicz-finetune")


def predict(text):
    return pipe(text)[0]["generated_text"]


demo = gr.Interface(
    fn=predict,
    inputs="text",
    outputs="text",
)

demo.launch()