Update app.py

app.py

@@ -7,7 +7,7 @@ from collections import Counter
 
 # --- Download model from HF Hub ---
 repo_id = "Ym420/Peptide-Function"  # replace with your HF repo
-model_filename = "xgb_multilabel_model_full.pkl"
+model_filename = "xgb_multilabel_model_full_extendedFuture.pkl"
 
 model_path = hf_hub_download(repo_id=repo_id, filename=model_filename)
 model_package = joblib.load(model_path)
@@ -18,12 +18,14 @@ feature_columns = model_package['feature_columns']
 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']
 
 # --- Target cells ---
 TARGET_CELLS = ["Gram+", "Fungus", "Mammalian Cell", "Cancer", "Gram-"]