Update app.py

app.py

@@ -16,18 +16,18 @@ model_package = joblib.load(model_path)
 model_dict = model_package['model']           # dict: {'Gram+': XGBClassifier, ...}
 feature_columns = model_package['feature_columns']
 
-# --- Metadata ---
-aa_list = model_package['aa_list']
-dipeptides = model_package['dipeptides']
-hydrophobicity_scale = model_package['hydrophobicity_scale']
-eisenberg_scale = model_package['eisenberg_scale']
-aa_mass = model_package['aa_mass']
-aa_charge = model_package['aa_charge']
-aa_boman = model_package['aa_boman']
-aa_flexibility = model_package['aa_flexibility']
-aa_polarizability = model_package['aa_polarizability']
-aa_aliphatic = model_package['aa_aliphatic']
-aa_deltaG = model_package['aa_deltaG']
+# --- Metadata (all restored) ---
+aa_list = model_package.get('aa_list', [])
+dipeptides = model_package.get('dipeptides', [])
+hydrophobicity_scale = model_package.get('hydrophobicity_scale', {})
+eisenberg_scale = model_package.get('eisenberg_scale', {})
+aa_mass = model_package.get('aa_mass', {})
+aa_charge = model_package.get('aa_charge', {})
+aa_boman = model_package.get('aa_boman', {})
+aa_flexibility = model_package.get('aa_flexibility', {})
+aa_polarizability = model_package.get('aa_polarizability', {})
+aa_aliphatic = model_package.get('aa_aliphatic', {})
+aa_deltaG = model_package.get('aa_deltaG', {})
 
 # --- Target cells ---
 TARGET_CELLS = ["Gram+", "Fungus", "Mammalian Cell", "Cancer", "Gram-"]
@@ -120,3 +120,8 @@ with gr.Blocks(css=custom_css, theme="default") as demo:
 
 if __name__ == "__main__": 
     demo.launch(show_error=True)
+
+
+
+
+