Update README.md

README.md

@@ -40,23 +40,42 @@ pip install peft sentence-transformers
 from sentence_transformers import SentenceTransformer
 from peft import PeftModel
 
-model = SentenceTransformer("all-mpnet-base-v2")
-model[0].auto_model = PeftModel.from_pretrained(model[0].auto_model, "vahidthegreat/StanceAware-SBERT")
+base_model = SentenceTransformer("all-mpnet-base-v2")
+finetuned_model = SentenceTransformer("all-mpnet-base-v2")
+finetuned_model[0].auto_model = PeftModel.from_pretrained(finetuned_model[0].auto_model, "vahidthegreat/StanceAware-SBERT")
 
 sentences = [
     "I love pineapple on pizza",
     "I hate pineapple on pizza",
     "I like pineapple on pizza",
 ]
-embeddings = model.encode(sentences)
+embeddings = base_model.encode(sentences)
+print('Embedding Shape:')
 print(embeddings.shape)
 
-similarity = model.similarity(embeddings, embeddings)
+similarity = base_model.similarity(embeddings, embeddings)
+print('\n\nSimilarity Matrix for the Base Model:')
+print(similarity)
+
+
+embeddings = finetuned_model.encode(sentences)
+similarity = finetuned_model.similarity(embeddings, embeddings)
+print('\n\nSimilarity Matrix for the FineTuned Model:')
 print(similarity)
 ```
 
 ```output
+Embedding Shape:
 (3, 768)
+
+
+Similarity Matrix for the Base Model:
+tensor([[1.0000, 0.8591, 0.9774],
+        [0.8591, 1.0000, 0.8561],
+        [0.9774, 0.8561, 1.0000]])
+
+
+Similarity Matrix for the FineTuned Model:
 tensor([[1.0000, 0.5732, 0.9713],
         [0.5732, 1.0000, 0.5804],
         [0.9713, 0.5804, 1.0000]])