Update app.py

app.py

@@ -5,6 +5,9 @@ from flair.models import TextClassifier
 from flair.data import Sentence
 import matplotlib.pyplot as plt
 
+# Load Flair model outside the function
+classifier = TextClassifier.load('en-sentiment')
+
 # Function to perform sentiment analysis using TextBlob model
 def textblob_sentiment(text):
     blob = TextBlob(text)
@@ -18,7 +21,6 @@ def vader_sentiment(text):
 
 # Function to perform sentiment analysis using Flair model
 def flair_sentiment(text):
-    classifier = TextClassifier.load('en-sentiment')
     sentence = Sentence(text)
     classifier.predict(sentence)
     if sentence.labels[0].value == 'POSITIVE':
@@ -50,3 +52,4 @@ ax.bar(['TextBlob', 'VADER', 'Flair'], [textblob_score, vader_score, flair_score
 ax.axhline(y=0, color='gray', linestyle='--')
 ax.set_title('Sentiment Scores')
 st.pyplot(fig)
+plt.close(fig)  # Close the figure to prevent memory leaks