Adding input text to be embedded

app.py

@@ -4,6 +4,9 @@ import pandas as pd
 from sentence_transformers import SentenceTransformer
 model = SentenceTransformer('paraphrase-MiniLM-L6-v2')
 
+input_sentence = st.text_input('Movie title', 'Life of Brian')
+#st.write('The current movie title is', title)
+
 #Sentences we want to encode. Example:
 sentence = ['This framework generates embeddings for each input sentence']
 
@@ -12,7 +15,8 @@ sentence = ['This framework generates embeddings for each input sentence']
 embedding = model.encode(sentence)
 
 x = st.slider('Select a value')
-st.write(x, 'squared is', x * x, 'embedding', embedding[0][0])
+#st.write(x, 'squared is', x * x, 'embedding', embedding[0][0])
+st.write('The embedding of', input_sentence, 'at position',x,'is',embedding[0][int(x)])
 
 
 uploaded_file = st.file_uploader("Choose a file")