Update app.py

app.py

@@ -6,11 +6,23 @@ import faiss
 import torch
 import json
 import spaces
+
 # Load CSV data
 data = pd.read_csv('RBD10kstats.csv')
 
-# Convert embedding column from JSON string to numpy array
-data['embedding'] = data['embedding'].apply(lambda x: np.array(json.loads(x)))
+# Function to safely convert JSON strings to numpy arrays
+def safe_json_loads(x):
+    try:
+        return np.array(json.loads(x))
+    except json.JSONDecodeError as e:
+        print(f"Error decoding JSON: {e}")
+        return np.array([])  # Return an empty array or handle it as appropriate
+
+# Apply the safe_json_loads function to the embedding column
+data['embedding'] = data['embedding'].apply(safe_json_loads)
+
+# Filter out any rows with empty embeddings
+data = data[data['embedding'].apply(lambda x: x.size > 0)]
 
 # Initialize FAISS index
 dimension = len(data['embedding'][0])