import gradio as gr
from transformers import pipeline

#Calling the pipeline for text generation
generator = pipeline('text-generation', model='FloVolo/mistral-flo-finetune-2-T4')

def generation_text(text):
    return generator(text, max_length=250, do_sample=False)[0]['generated_text']


demo = gr.Interface(
    fn=generation_text,
    inputs=["text"],
    outputs=["text"],
)

demo.launch()