Update app.py

app.py

@@ -105,10 +105,10 @@ class HybridHeadModel(nn.Module):
 
 # --- Model Catalog ---
 MODEL_CATALOG = {
-    "AnatomyFlaws-v15.5 (DINOv3 7b 8-bit)": {
+    "AnatomyFlaws-v15.5 (DINOv3 7b int4)": {
         "repo_id": "Enferlain/lumi-classifier",
         "config_filename": "AnatomyFlaws-v15.5_dinov3_7b_bnb_fl.config.json",
-        "head_filename": "AnatomyFlaws-v15.5_dinov3_7b_bnb_fl_s4K.safetensors",
+        "head_filename": "AnatomyFlaws-v15.5_dinov3_7b_bnb_fl_s3K_best_val.safetensors",
         # Explicitly define the vision model repo ID to prevent errors
         # "vision_model_repo_id": "Enferlain/dinov3-vit7b16-pretrain-lvd1689m-8bit"
         "vision_model_repo_id": "Enferlain/dinov3-vit7b16-pretrain-lvd1689m-int4",
@@ -257,6 +257,7 @@ def predict_anatomy_v3(image: Image.Image, model_name: str):
 DESCRIPTION = """
 ## Lumi's Anatomy Flaw Classifier Demo ✨
 Select a model from the dropdown, then upload an image to classify its anatomy/structural correctness.
+Might be slow and quality might be off, has to be int4 cuz bnb8bit needs cuda and the demo runs on cpu.
 """
 EXAMPLE_DIR = "examples"