AdeelHassan commited on
Commit
c0c33df
·
verified ·
1 Parent(s): c317afe
Files changed (1) hide show
  1. inference.py +1 -1
inference.py CHANGED
@@ -123,7 +123,7 @@ def _load_hf_models() -> None:
123
  # GAN-focused: ViT fine-tuned on GAN / deepfake face datasets
124
  _model_2 = pipeline(
125
  "image-classification",
126
- model="Wvolf/ViT_Deepfake_Detection",
127
  device=hf_device,
128
  )
129
  except Exception:
 
123
  # GAN-focused: ViT fine-tuned on GAN / deepfake face datasets
124
  _model_2 = pipeline(
125
  "image-classification",
126
+ model="mikedata/real_vs_fake_image_model_vit_base",
127
  device=hf_device,
128
  )
129
  except Exception: