Spaces:
Runtime error
Runtime error
import gradio as gr | |
from transformers import pipeline | |
model_id = "knkarthick/MEETING_SUMMARY" | |
generator = pipeline(task="text2text-generation", model=model_id) | |
#def split_paragraph(paragraph, max_chunk_size=1024): | |
# words = paragraph.split() | |
# chunks = [] | |
# current_chunk = [] | |
# current_chunk_size = 0 | |
# for word in words: | |
# word_len = len(word) + 1 # Add 1 for the space | |
# if current_chunk_size + word_len <= max_chunk_size: | |
# current_chunk.append(word) | |
# current_chunk_size += word_len | |
# else: | |
# chunks.append(' '.join(current_chunk)) | |
# current_chunk = [word] | |
# current_chunk_size = word_len | |
# if current_chunk: | |
# chunks.append(' '.join(current_chunk)) | |
# return chunks | |
def launch(input): | |
# if len(input) > 1024: | |
# return " ".join([res["generated_text"] for res in generator(split_paragraph(input))]) | |
return generator(input)[0]["generated_text"] | |
iface = gr.Interface(launch, inputs="text", outputs="text") | |
iface.launch() |