test bug fix

app.py

@@ -2,7 +2,7 @@ import streamlit as st
 from datasets import load_dataset
 dataset = load_dataset("roneneldan/TinyStories")
 
-st.write(dataset['train'][0]['text'])
+st.write(dataset['train'][10]['text'])
 
 #-------------------------------------------------------------
 #-------------------------------------------------------------
@@ -12,9 +12,10 @@ model = SentenceTransformer('all-MiniLM-L6-v2')
 
 # Sentences from the data set
 sentences = [item['text'] for item in dataset['train'][:10]]
+sentences = [dataset['train'][0],dataset['train'][1],dataset['train'][2]]
 
 #Compute embedding 
-embeddings1 = model.encode(sentences, convert_to_tensor=True)
+embeddings = model.encode(sentences, convert_to_tensor=True)
 
 #Compute cosine-similarities
 cosine_scores = util.cos_sim(embeddings, embeddings)