BrandSight / src /mistral_llm.py
akashjayampu's picture
Create mistral_llm.py
bcb51a3 verified
raw
history blame
269 Bytes
from transformers import pipeline
summarizer = pipeline("summarization", model="sshleifer/distilbart-cnn-6-6")
def analyze_text(text):
try:
return summarizer(text[:1024])[0]["summary_text"]
except Exception as e:
return f"Summary failed: {e}"