Update train.py

train.py

@@ -19,7 +19,7 @@ plt.rcParams.update({
 })
 
 # Load dataset
-data = pd.read_csv("/storage/group/cdm8/default/BindPred/embeddings/Seq_Gen_updated.csv")
+data = pd.read_csv("/storage/group/cdm8/default/BindPred/embeddings/xxx.csv")
 
 # Handle missing values
 data['Ligand Features'] = data['Ligand Features'].fillna('')