Update app.py

app.py

@@ -186,6 +186,10 @@ def calculate_shap_fn(texts, model, cfg):
 
 
 def calculate_shap_values_core(model_choice, organism_choice, sequences, cfg=Config()):
+    cell_line = "HeLa" if organism_choice == "Human" else "NIH3T3"
+    cfg.model = f"sagawa/PLTNUM-{model_choice}-{cell_line}"
+    cfg.architecture = model_choice
+    cfg.model_path = f"sagawa/PLTNUM-{model_choice}-{cell_line}"
     cfg.device = "cuda" if torch.cuda.is_available() else "cpu"
 
     seed_everything(cfg.seed)