Update app.py
Browse files
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].
|
| 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}")
|