Update app.py

app.py

@@ -1,4 +1,5 @@
 import streamlit as st
+import numpy as np
 from sentence_transformers import SentenceTransformer
 
 # Load the pre-trained model
@@ -13,5 +14,7 @@ if sentences:
     # Split sentences by new line
     sentences_list = [s.strip() for s in sentences.split('\n') if s.strip()]
     # Get embeddings
-    embeddings = model.encode(sentences_list).tolist()
-    st.json(embeddings)
+    embeddings = model.encode(sentences_list)
+    # Convert to 2D NumPy array
+    embeddings_array = np.array(embeddings)
+    st.write(embeddings_array)