aneela-pervez commited on
Commit
5b95014
·
verified ·
1 Parent(s): 4195790

Update fake_image.py

Browse files
Files changed (1) hide show
  1. 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", repo_type="model", force_download=True))
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()):