File size: 312 Bytes
1091b7f
 
 
 
 
 
b2c9c78
1091b7f
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
from transformers import pipeline
import gradio as gr


model = pipeline("summarization")

def predict(prompt: str) -> str:
    summary = model(prompt)[0]["summary_text"]
    return summary


# create an interface for the model
with gr.Interface(predict, "textbox", "text") as interface:
    interface.launch()