Spaces:
Sleeping
Sleeping
# now to create an interface with gradio | |
import gradio as gr | |
from transformers import pipeline | |
from gradio_space_ci import enable_space_ci | |
enable_space_ci() | |
pipe = pipeline("text2text-generation", model="google/flan-t5-base") | |
# a function that uses the pipeline | |
# takes text as input and passes it to the pipeline | |
def chat(text): | |
output = pipe(text)[0]["generated_text"] | |
return output | |
# fucntion , input of type text , output of type text | |
demo = gr.Interface(chat , "text" , "text" ) | |
demo.launch() |