Update app.py

app.py

@@ -124,7 +124,7 @@ class HybridHeadModel(nn.Module):
 # Replace with your HF username and repo name
 HUB_REPO_ID = "Enferlain/lumi-classifier" # Or wherever you uploaded the model
 # Use the specific checkpoint you want (e.g., s9k or the best_val one)
-HEAD_MODEL_FILENAME = "AnatomyFlaws-v11.3_adabelief_fl_naflex_3000_s9K.safetensors"
+HEAD_MODEL_FILENAME = "AnatomyFlaws-v11.3_adabelief_fl_naflex_3000_s6K_best_val.safetensors"
 # Usually config corresponds to the base run name, not a specific step
 CONFIG_FILENAME = "AnatomyFlaws-v11.3_adabelief_fl_naflex_3000.config.json"
 
@@ -262,7 +262,6 @@ DESCRIPTION = """
 Upload an image to classify its anatomy as 'Good' or 'Bad'.
 This model uses embeddings from **google/siglip2-so400m-patch16-naflex**
 and a custom **HybridHeadModel** fine-tuned for anatomy classification.
-Model Checkpoint: **AnatomyFlaws-v11.3_..._s9K** (or specify which one).
 """
 
 # Add example images if you have some in an 'examples' folder in the Space repo