Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -253,7 +253,7 @@ def run_diagnosis(
|
|
| 253 |
|
| 254 |
# Load model (cache for speed)
|
| 255 |
if backbone_name not in MODEL_CACHE:
|
| 256 |
-
print(f"Loading model weights from {ckpt_path")
|
| 257 |
MODEL_CACHE[backbone_name] = modelModule.load_from_checkpoint(
|
| 258 |
ckpt_path, backbone_model_name=backbone_info["torchvision_name"])
|
| 259 |
model = MODEL_CACHE[backbone_name]
|
|
|
|
| 253 |
|
| 254 |
# Load model (cache for speed)
|
| 255 |
if backbone_name not in MODEL_CACHE:
|
| 256 |
+
print(f"Loading model weights from {ckpt_path}")
|
| 257 |
MODEL_CACHE[backbone_name] = modelModule.load_from_checkpoint(
|
| 258 |
ckpt_path, backbone_model_name=backbone_info["torchvision_name"])
|
| 259 |
model = MODEL_CACHE[backbone_name]
|