app.py

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()