Update app.py

app.py

@@ -3,19 +3,13 @@ import pickle
 import os
 from datasets import load_dataset
 dataset = load_dataset("SandipPalit/Movie_Dataset")
-print(os.getcwd())
-pickle.dump(dataset,open('./df.pkl','wb'))
-temp=pickle.load(open('./df.pkl','rb'))
-print(temp['train']['Title'])
 from InstructorEmbedding import INSTRUCTOR
 model = INSTRUCTOR('hkunlp/instructor-xl')
 
 from sklearn.metrics.pairwise import cosine_similarity
 
 def getSimilarity(sentences_a,sentences_b):
-  print(sentences_a)
-  embeddings_a = model.encode(sentences_a)
-  print(embeddings_a)
+  embeddings_a = pickle.load(open(os.getcwd(),'rb'))
   embeddings_b = model.encode(sentences_b)
   similarities = cosine_similarity(embeddings_a,embeddings_b)
   return similarities