Spaces:
Runtime error
Runtime error
from transformers import pipeline | |
import gradio as gr | |
#model | |
model = pipeline("summarization", model="sshleifer/distilbart-cnn-12-6") #model="sshleifer/distilbart-cnn-12-6") | |
def func(Text): | |
words = Text.split() | |
grouped_words = [' '.join(words[i: i + 300]) for i in range(0, len(words), 300)] | |
summary_list = [] | |
for i in grouped_words: | |
y = model(i, max_length=80, do_sample=False) | |
summary_list.append(y) | |
final_summary = ','.join(str(v) for v in summary_list) | |
return final_summary | |
import gradio as gr | |
app = gr.Interface(fn=func, inputs="textbox", outputs="textbox", title="InfluencerAI-Summarizer") | |
app.launch() |