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