ceejaytheanalyst commited on
Commit
31e5aac
·
verified ·
1 Parent(s): 4174702

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -11,7 +11,7 @@ import pickle
11
  #pipeline = pipeline(task="Sentence Similarity", model="all-MiniLM-L6-v2")
12
  model = SentenceTransformer('neuml/pubmedbert-base-embeddings')
13
  #sentence_embed = pd.read_csv('Reference_file.csv')
14
- with open("embeddings.pkl", "rb") as fIn:
15
  stored_data = pickle.load(fIn)
16
  stored_code = stored_data["SBS_code"]
17
  stored_sentences = stored_data["sentences"]
 
11
  #pipeline = pipeline(task="Sentence Similarity", model="all-MiniLM-L6-v2")
12
  model = SentenceTransformer('neuml/pubmedbert-base-embeddings')
13
  #sentence_embed = pd.read_csv('Reference_file.csv')
14
+ with open("embeddings_1.pkl", "rb") as fIn:
15
  stored_data = pickle.load(fIn)
16
  stored_code = stored_data["SBS_code"]
17
  stored_sentences = stored_data["sentences"]