xljesse commited on
Commit
2120080
·
verified ·
1 Parent(s): ff6cd57

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -231,7 +231,7 @@ class AudioDeepfakeDetector:
231
  cache_dir="./models"
232
  )
233
  ckpt = torch.load(checkpoint_path, map_location=self.device)
234
- self.models[model_type].load(ckpt)
235
  print(f"✅ Loaded checkpoint for {model_type}")
236
  except Exception as e:
237
  print(f"⚠️ Could not load checkpoint for {model_type}: {e}")
 
231
  cache_dir="./models"
232
  )
233
  ckpt = torch.load(checkpoint_path, map_location=self.device)
234
+ self.models[model_type].load_state_dict(ckpt)
235
  print(f"✅ Loaded checkpoint for {model_type}")
236
  except Exception as e:
237
  print(f"⚠️ Could not load checkpoint for {model_type}: {e}")