Update app.py

app.py

@@ -1,14 +1,3 @@
-# appgradiofinal3_radio.py
-# Gradio — TBNet + Lung U-Net Auto Mask + Grad-CAM + RADIO
-# + SAFER PHONE MODE + MASK POST-PROCESSING + MASK SANITY FAILSAFE
-# + 3-STATE CONSENSUS (LOW / INDET / SCREEN+)
-#
-# Run:
-#   python appgradiofinal3_radio.py
-#
-# Requirements:
-#   pip install gradio timm torchvision opencv-python pillow transformers einops
-
 import os
 import cv2
 import numpy as np
@@ -28,20 +17,22 @@ from PIL import Image
 
 
 # ============================================================
-# USER CONFIG
+# USER CONFIG (HF Spaces friendly)
 # ============================================================
 
 # ---- Default TB/Lung weights ----
 DEFAULT_TB_WEIGHTS = "weights/best.pt"
-DEFAULT_LUNG_WEIGHTS = "lung_unet_mont_shenzhen.pt"
+DEFAULT_LUNG_WEIGHTS = "weights/lung_unet_mont_shenzhen.pt"
 
 # ---- RADIO config (same env as TB) ----
 RADIO_HF_REPO = "nvidia/C-RADIOv4-SO400M"
 RADIO_REVISION = "c0457f5dc26ca145f954cd4fc5bb6114e5705ad8"
 
+# Your trained heads stored in this Space repo
 RADIO_RAW_HEAD_PATH = "weights/best_raw.pt"
 RADIO_MASKED_HEAD_PATH = "weights/best_masked.pt"
 
+
 RADIO_IMG_SIZE = 320
 RADIO_PATCH_SIZE = 16
 RADIO_THR_SCREEN = 0.05