Update prediction.py

prediction.py

@@ -28,12 +28,12 @@ def load_chemberta():
 
 # Load scalers
 scalers = {
-    "Tensile Strength": joblib.load("scaler_Tensile_strength_Mpa_.joblib"),
-    "Ionization Energy": joblib.load("scaler_Ionization_Energy_eV_.joblib"),
-    "Electron Affinity": joblib.load("scaler_Electron_Affinity_eV_.joblib"),
+    "Tensile Strength(Mpa)": joblib.load("scaler_Tensile_strength_Mpa_.joblib"),
+    "Ionization Energy(eV)": joblib.load("scaler_Ionization_Energy_eV_.joblib"),
+    "Electron Affinity(eV)": joblib.load("scaler_Electron_Affinity_eV_.joblib"),
     "logP": joblib.load("scaler_LogP.joblib"),
     "Refractive Index": joblib.load("scaler_Refractive_Index.joblib"),
-    "Molecular Weight": joblib.load("scaler_Molecular_Weight_g_mol_.joblib")
+    "Molecular Weight(g/mol)": joblib.load("scaler_Molecular_Weight_g_mol_.joblib")
 }
 
 # Transformer model