Update app.py

app.py

@@ -1,5 +1,5 @@
 import streamlit as st
-from sentence_transformers import SentenceTransformer
+from sentence_transformers import SentenceTransformer, util
 
 # Load the model
 model = SentenceTransformer('sentence-transformers/msmarco-distilbert-dot-v5')
@@ -24,8 +24,7 @@ def main():
 
 # Function to get the embedding
 def get_emb(text):
-    text_emb = model.encode(text)
-    return text_emb
+    return model.encode(text)
 
 # Run the Streamlit app
 if __name__ == "__main__":