case_study / utils.py
GurgenGulay's picture
Update utils.py
e854524 verified
raw
history blame
804 Bytes
from transformers import pipeline
# Create the pipeline globally
pipe = pipeline("text2text-generation", model="google-t5/t5-base", device="cpu")
def split_text_into_chunks(text, chunk_size=1000):
words = text.split()
chunks = []
for i in range(0, len(words), chunk_size):
chunk = ' '.join(words[i:i+chunk_size])
chunks.append(chunk)
return chunks
def generate_lesson_from_chunks(chunks):
generated_texts = []
for chunk in chunks:
generated_text = pipe(chunk, max_length=500)[0]['generated_text']
generated_texts.append(generated_text)
return ' '.join(generated_texts)
def process_large_text(text):
chunks = split_text_into_chunks(text, chunk_size=1000)
generated_text = generate_lesson_from_chunks(chunks)
return generated_text