Update main.py

main.py

@@ -9,10 +9,11 @@ drive.mount('/content/drive')
 
 from transformers import pipeline
 
-# Load pre-trained model for emotion detection
+# Initialize the emotion classifier pipeline globally
 emotion_classifier = pipeline("text-classification", model="bhadresh-savani/distilbert-base-uncased-emotion")
 
 def detect_emotion(text):
+    # Ensure the emotion_classifier is used properly
     result = emotion_classifier(text)
     emotion = result[0]['label']
     confidence = result[0]['score']
@@ -87,6 +88,7 @@ def generate_emotional_speech(text, emotion):
 from IPython.display import Audio, display
 
 def emotion_aware_tts_pipeline(text):
+    # Ensure the emotion_classifier is being accessed globally
     emotion, confidence = detect_emotion(text)
     print(f"Emotion Detected: {emotion} with Confidence: {confidence:.2f}")