Update app.py

app.py

@@ -319,7 +319,7 @@ def run_diagnosis(
         raise FileNotFoundError(f"Checkpoint not found: {ckpt_path}")
 
     # Load classification model (cache for speed)
-    if backbone_name not in MODEL_CACHE:
+    if backbone_name not in ViT_MODEL_CACHE:
       ViT_MODEL_CACHE[backbone_name] = modelModule.load_from_checkpoint(
         ckpt_path, backbone_model_name=backbone_name, num_layers_to_unfreeze = 2)
     model = ViT_MODEL_CACHE[backbone_name]