speaker2 / agent /planner.py
subramaniansrc's picture
Update agent/planner.py
acc0700 verified
Raw
History Blame Contribute Delete
803 Bytes
from transformers import pipeline
from transformers import pipeline
generator = pipeline(
"text-generation",
model="google/flan-t5-base"
)
def explain_slide(text):
if not text.strip():
return "This slide does not contain readable content."
prompt = f"""
Explain the following slide content clearly for engineering students.
Slide:
{text}
Explanation:
"""
result = generator(prompt, max_new_tokens=120)[0]["generated_text"]
return result
)
def explain_slide(text):
if not text.strip():
return "This slide does not contain readable content."
prompt = f"""
You are a university professor.
Explain this slide clearly to engineering students.
Slide:
{text}
Explanation:
"""
result = generator(prompt)[0]["generated_text"]
return result