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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
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['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['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)