import gradio as gr from transformers import pipeline pipe = pipeline('text-generation', model='facebook/opt-350m') input = gr.Interface.TextBox() output = gr.Interface.TextBox() intf = gr.Interface(fn = pipe, inputs = input, outputs = output) intf.launch()