Update app.py

app.py

@@ -2,6 +2,8 @@ from transformers import pipeline
 import matplotlib.pyplot as plt
 import streamlit as st
 from PIL import Image
+import pandas as pd
+import numpy as np
 
 
 
@@ -56,6 +58,7 @@ if file_name is not None:
     plt.tight_layout()
     plt.show()
 
+
     # 识别每个人的表情
     output_list = []
     
@@ -80,8 +83,25 @@ if file_name is not None:
     labels = list(label_counts.keys())
     sizes = list(label_counts.values())
     
-    plt.figure(figsize=(8, 8))
-    plt.pie(sizes, labels=labels, autopct='%1.1f%%', startangle=140)
-    plt.title('Distribution of Labels')
-    plt.axis('equal')  # 确保饼状图为圆形
-    plt.show()
+    fig, ax = plt.subplots()
+    fig.pie(sizes, labels=labels, autopct='%1.1f%%', startangle=140)
+    fig.title('Distribution of Labels')
+    fig.axis('equal')  # 确保饼状图为圆形
+    # plt.show()
+
+        
+    # chart_data = pd.DataFrame(
+    #     {
+    #         "col1": list(range(20)),
+    #         "col2": np.random.randn(20),
+    #         "col3": np.random.randn(20),
+    #     }
+    # )
+    
+    # st.bar_chart(
+    #     chart_data,
+    #     x="col1",
+    #     y=["col2", "col3"],
+    #     color=["#FF0000", "#0000FF"],  # Optional
+    # )
+    st.pyplot(fig)