Update app.py

app.py

@@ -1,5 +1,5 @@
 import streamlit as st
-from transformers import AutoModel, BertTokenizer
+from transformers import BertModel, BertTokenizer
 from sklearn.decomposition import PCA
 import plotly.graph_objs as go
 import numpy as np
@@ -7,7 +7,7 @@ from database_utils import init_db, save_embeddings_to_db, get_all_embeddings, c
 
 # Initialize BERT model and tokenizer
 tokenizer = BertTokenizer.from_pretrained('bert-base-uncased')
-model = AutoModel.from_pretrained('bert-base-uncased')
+model = BertModel.from_pretrained('bert-base-uncased')
 
 def get_bert_embeddings(words):
     embeddings = []
@@ -83,8 +83,8 @@ def main():
         st.download_button(label="Download CSV", data=csv, file_name='embeddings.csv', mime='text/csv')
 
     embeddings, words = get_all_embeddings()
-    embeddings = np.array(embeddings)
-    if embeddings.size > 0:
+    if len(embeddings) > 0:
+        embeddings = np.array(embeddings)
         plot_interactive_bert_embeddings(embeddings, words)
 
 if __name__ == "__main__":