Spaces:
Runtime error
Runtime error
ok
Browse files- 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="
|
| 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:
|