akashjayampu commited on
Commit
bcb51a3
·
verified ·
1 Parent(s): b4feefb

Create mistral_llm.py

Browse files
Files changed (1) hide show
  1. src/mistral_llm.py +9 -0
src/mistral_llm.py ADDED
@@ -0,0 +1,9 @@
 
 
 
 
 
 
 
 
 
 
1
+ from transformers import pipeline
2
+
3
+ summarizer = pipeline("summarization", model="sshleifer/distilbart-cnn-6-6")
4
+
5
+ def analyze_text(text):
6
+ try:
7
+ return summarizer(text[:1024])[0]["summary_text"]
8
+ except Exception as e:
9
+ return f"Summary failed: {e}"