Update fake_image.py
Browse files- fake_image.py +1 -1
fake_image.py
CHANGED
|
@@ -59,7 +59,7 @@ model = Deepfake_Hybrid_Network(num_classes=2).to(device)
|
|
| 59 |
|
| 60 |
try:
|
| 61 |
# Model HF repo se download ho raha hy
|
| 62 |
-
model_path = hf_hub_download(repo_id="aneela-pervez/My-Deepfake-Models", filename="Deepfake_Hybrid_SOTA.pth"
|
| 63 |
state_dict = torch.load(model_path, map_location=device)
|
| 64 |
|
| 65 |
if any(k.startswith('module.') for k in state_dict.keys()):
|
|
|
|
| 59 |
|
| 60 |
try:
|
| 61 |
# Model HF repo se download ho raha hy
|
| 62 |
+
model_path = hf_hub_download(repo_id="aneela-pervez/My-Deepfake-Models", filename="Deepfake_Hybrid_SOTA.pth")
|
| 63 |
state_dict = torch.load(model_path, map_location=device)
|
| 64 |
|
| 65 |
if any(k.startswith('module.') for k in state_dict.keys()):
|