Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -10,8 +10,7 @@ device = 'cuda' if torch.cuda.is_available() else 'cpu'
|
|
10 |
|
11 |
# Load the CSV file with embeddings
|
12 |
df = pd.read_csv('RBDx10kstats.csv')
|
13 |
-
df['
|
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)
|
|
|
10 |
|
11 |
# Load the CSV file with embeddings
|
12 |
df = pd.read_csv('RBDx10kstats.csv')
|
13 |
+
df['embedding'] = df['embedding'].apply(json.loads) # Convert JSON string back to list
|
|
|
14 |
|
15 |
# Convert embeddings to tensor for efficient retrieval
|
16 |
embeddings = torch.tensor(df['embedding'].tolist(), device=device)
|