Spaces:
Runtime error
Runtime error
File size: 410 Bytes
d76c37d |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 |
import gradio as gr
from transformers import pipeline
import torch
torch.set_default_dtype(torch.float32)
summarizer = pipeline("summarization", model="facebook/bart-large-cnn")
print("Model Loaded")
def summarize(text):
return (summarizer(text, max_length=300, min_length=200, do_sample=False, num_workers=4))
iface = gr.Interface(fn=summarize, inputs="text", outputs="text")
iface.launch(debug=True)
|