add threshold for adjacency matrix

Add threshold for similarity matrix. TODO: add slider, remove diagonal

app.py

@@ -14,7 +14,9 @@ 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]]
+#sentences = [dataset['train'][ii] for ii in range(10)]
 
 #Compute embedding 
 embeddings = model.encode(sentences, convert_to_tensor=True)
@@ -33,7 +35,7 @@ for i in range(len(sentences)):
         A[j][i] = cosine_scores[i][j]
 
 #G = nx.from_numpy_array(A)
-G = nx.from_numpy_array(cosine_scores.numpy())
+G = nx.from_numpy_array(cosine_scores.numpy()>0.4)
 
 
 #-------------------------------------------------------------