Update app.py

app.py

@@ -81,8 +81,8 @@ def analyze_emotions_in_comments(df: pd.DataFrame) -> tuple:
     model = load_model()
     selected_columns = ['text', 'author', 'published_at']
     df = df[selected_columns]
-    res_list = []    
-    res_list = model(df['text'].to_list())       
+    res_list = []
+    res_list = model(df['text'].to_list()) 
     full_df = pd.concat([pd.DataFrame(res_list), df], axis=1)
     return (full_df, len(res_list))
 
@@ -114,7 +114,7 @@ def visualize_data(df: pd.DataFrame):
     Visualizes the data from the input DataFrame and returns a matplotlib figure object.
     Args: df (DataFrame): The input DataFrame containing the data to be visualized.
     Returns: fig: A matplotlib figure object
-    """    
+    """
     data = df['label'].value_counts()
     fig, ax = plt.subplots()
     plt.title("Эмоциональная окраска комментариев на YouTube")
@@ -147,8 +147,8 @@ if st.session_state.start:
     st.markdown('***')
 
     # Выводим heatmap комментариев по часам и датам
-    st.pyplot(plot_heatmap_from_dataframe(full_df)) 
+    st.pyplot(plot_heatmap_from_dataframe(full_df))
     st.markdown('***')
 
-    # Выводим круговую диаграмму   
+    # Выводим круговую диаграмму
     st.pyplot(visualize_data(full_df))