File size: 417 Bytes
72ffc90 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 |
import gradio as gr
from transformers import pipeline
pipe = pipeline("text-generation", model="gpt2")
# 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()
|