Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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['
|
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)
|