File size: 396 Bytes
30fd011 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 |
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()
|