Update app.py

app.py

@@ -25,14 +25,15 @@ if submit_button:
 
        # Perform calculations
        
-       sentence_embeddings = model.encode(sentences)
-
+       # Append input sentences to 'sentences' list
        sentences.append(sentence_1)
        sentences.append(sentence_2)
        
-       for sentence, embedding in zip(sentences, sentence_embeddings):
-           st.write("Sentence:", sentence)
-           st.write("Embedding:", embedding)
-           st.write("")
+       # Create embeddings for both sentences
+       sentence_embeddings = model.encode(sentences)
+      
               
-
+       st.write('Similarity between {} and {} is {}'.format(sentence_1,
+              sentence_2,
+              cosine_similarity(sentence_embeddings[0].reshape(1, -1),
+              sentence_embeddings[1].reshape(1, -1))[0][0]))