Update eda.py

eda.py

@@ -18,7 +18,7 @@ def label_sentiment(rating):
 def process_review(review):
     """Simple processing for the review text."""
     review = review.lower()
-    review = re.sub(r'[^a-z\s]', '', review) # Remove non-alphabetical characters
+    review = re.sub(r'[^a-z\s]', '', review)  # Remove non-alphabetical characters
     return review
 
 def display_eda(data):
@@ -48,7 +48,7 @@ def display_eda(data):
         subset = data[data['sentiment'] == sentiment]
         text = " ".join(process_review(review) for review in subset['review_description'])
         wordcloud = WordCloud(max_words=100, background_color="white").generate(text)
-        plt.figure()
+        fig = plt.figure()
         plt.imshow(wordcloud, interpolation="bilinear")
         plt.axis("off")
-        st.pyplot()
+        st.pyplot(fig)