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 |
|