Yoxas commited on
Commit
e56dc84
·
verified ·
1 Parent(s): 3fc9b48

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -14,7 +14,7 @@ df['Abstract'] = df['Abstract']
14
  #.apply(json.loads) # Convert JSON string back to list
15
 
16
  # Convert embeddings to tensor for efficient retrieval
17
- embeddings = torch.tensor(df['Abstract'].tolist(), device=device)
18
 
19
  # Load the Sentence Transformer model
20
  model = SentenceTransformer('all-MiniLM-L6-v2', device=device)
 
14
  #.apply(json.loads) # Convert JSON string back to list
15
 
16
  # Convert embeddings to tensor for efficient retrieval
17
+ embeddings = torch.tensor(df['embedding'].tolist(), device=device)
18
 
19
  # Load the Sentence Transformer model
20
  model = SentenceTransformer('all-MiniLM-L6-v2', device=device)