Update app.py

app.py

@@ -24,8 +24,12 @@ 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)]
 
+# Check if the DataFrame is empty after filtering
+if data.empty:
+    raise RuntimeError("No valid embeddings found in the data.")
+
 # Initialize FAISS index
-dimension = len(data['embedding'][0])
+dimension = len(data['embedding'].iloc[0])
 res = faiss.StandardGpuResources()  # use a single GPU
 
 # Check available GPU devices