AA-6055 commited on
Commit
d9bf1b2
·
verified ·
1 Parent(s): e3162c0

Create utils.py

Browse files
Files changed (1) hide show
  1. utils.py +35 -0
utils.py ADDED
@@ -0,0 +1,35 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ from transformers import pipeline
2
+ from gtts import gTTS
3
+
4
+ # Loading models
5
+ summarizer = pipeline("summarization", model="facebook/bart-large-cnn") # Load summarizer
6
+ sentiment_analyzer = pipeline("sentiment-analysis") # Load sentiment analyzer
7
+ # classifier = pipeline("zero-shot-classification", model="facebook/bart-large-mnli") # Load classifier
8
+
9
+ def analyze_sentiment(text):
10
+ result = sentiment_analyzer(text[:500])[0]
11
+ return result['label']
12
+
13
+ def summarize_text(text):
14
+ cleaned_text = text.strip().replace("\n", " ")
15
+ cleaned_text = cleaned_text[:3000] # Limit to avoid token overflow
16
+
17
+ result = summarizer(
18
+ cleaned_text,
19
+ max_length=130,
20
+ min_length=30,
21
+ do_sample=False
22
+ )
23
+
24
+ summary_text = result[0]['summary_text']
25
+ return summary_text
26
+
27
+ def generate_hindi_tts(text, filename="output.mp3"):
28
+ try:
29
+ tts = gTTS(text=text, lang='hi')
30
+ tts.save(filename)
31
+ print(f"Hindi audio saved to {filename}")
32
+ return filename
33
+ except Exception as e:
34
+ print(f"Error in generating the TTS: {e}")
35
+ return None