Hugging Face's logo

Spaces:
cfoli
/
ChestVision
Sleeping

App Files Community
cfoli commited on
Commit
35d3170
·
1 Parent(s): edab602

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -253,7 +253,8 @@ def run_diagnosis(
253
 
254
  # Load model (cache for speed)
255
  if backbone_name not in MODEL_CACHE:
256
- MODEL_CACHE[backbone_name] = modelModule.load_from_checkpoint(
 
257
  ckpt_path, backbone_model_name=backbone_info["torchvision_name"])
258
  model = MODEL_CACHE[backbone_name]
259
 
 
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]
260