case_study / text_analysis.py
GurgenGulay's picture
Update text_analysis.py
da75edc verified
raw
history blame
642 Bytes
from transformers import pipeline
pipe = pipeline("text2text-generation", model="google-t5/t5-base", device="cpu")
pipe.model.config.pad_token_id = pipe.tokenizer.eos_token_id
def generate_lesson_from_transcript(doc_text):
try:
generated_text = pipe(doc_text, max_length=100, truncation=True)[0]['generated_text']
output_path = "/tmp/generated_output.txt"
with open(output_path, "w") as file:
file.write(generated_text)
return generated_text, output_path
except Exception as e:
print(f"Bir hata oluştu: {str(e)}")
return "Bir hata oluştu", None