Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -105,7 +105,7 @@ class HybridHeadModel(nn.Module):
|
|
105 |
|
106 |
# --- Model Catalog ---
|
107 |
MODEL_CATALOG = {
|
108 |
-
"AnatomyFlaws-v15.5 (DINOv3 7b
|
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 |
-
|
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 |
|