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}")
|