Enferlain commited on
Commit
53532e2
·
verified ·
1 Parent(s): 105b065

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -105,7 +105,7 @@ class HybridHeadModel(nn.Module):
105
 
106
  # --- Model Catalog ---
107
  MODEL_CATALOG = {
108
- "AnatomyFlaws-v15.5 (DINOv3 7b int4)": {
109
  "repo_id": "Enferlain/lumi-classifier",
110
  "config_filename": "AnatomyFlaws-v15.5_dinov3_7b_bnb_fl.config.json",
111
  "head_filename": "AnatomyFlaws-v15.5_dinov3_7b_bnb_fl_s3K_best_val.safetensors",
@@ -286,7 +286,7 @@ def predict_anatomy_v3(image: Image.Image, model_name: str):
286
  DESCRIPTION = """
287
  ## Lumi's Anatomy Flaw Classifier Demo ✨
288
  Select a model from the dropdown, then upload an image to classify its anatomy/structural correctness.
289
- Might be slow and quality might be off, has to be int4 cuz bnb8bit needs cuda and the demo runs on cpu.
290
  """
291
  EXAMPLE_DIR = "examples"
292
 
 
105
 
106
  # --- Model Catalog ---
107
  MODEL_CATALOG = {
108
+ "AnatomyFlaws-v15.5 (DINOv3 7b bf16)": {
109
  "repo_id": "Enferlain/lumi-classifier",
110
  "config_filename": "AnatomyFlaws-v15.5_dinov3_7b_bnb_fl.config.json",
111
  "head_filename": "AnatomyFlaws-v15.5_dinov3_7b_bnb_fl_s3K_best_val.safetensors",
 
286
  DESCRIPTION = """
287
  ## Lumi's Anatomy Flaw Classifier Demo ✨
288
  Select a model from the dropdown, then upload an image to classify its anatomy/structural correctness.
289
+ Will be slow since it runs on cpu, ~2minutes on dinov3.
290
  """
291
  EXAMPLE_DIR = "examples"
292