File size: 344 Bytes
2b75707
7a09717
2b75707
 
7a09717
2b75707
 
 
 
7a09717
2b75707
 
1
2
3
4
5
6
7
8
9
10
11
12
import gradio as gr
# Use a pipeline as a high-level helper
from transformers import pipeline

pipe = pipeline("text-generation", model="mosaicml/mpt-7b-instruct", trust_remote_code=True)

def greet(name):
    return pipe('who is the president of USA')

# greet('df')
iface = gr.Interface(fn=greet, inputs="text", outputs="text")
iface.launch()