Spaces:
Runtime error
Runtime error
no log
Browse files- inference.py +4 -4
inference.py
CHANGED
|
@@ -115,26 +115,26 @@ def _load_hf_models() -> None:
|
|
| 115 |
|
| 116 |
if _model_hf1 is None:
|
| 117 |
try:
|
| 118 |
-
|
| 119 |
_model_hf1 = hf_pipeline(
|
| 120 |
"image-classification",
|
| 121 |
model="prithivMLmods/Deepfake-Detection-Exp-02-21",
|
| 122 |
device=hf_device,
|
| 123 |
)
|
| 124 |
-
|
| 125 |
except Exception as exc:
|
| 126 |
logger.warning(f"HF model 1 failed to load: {exc}")
|
| 127 |
_model_hf1 = False
|
| 128 |
|
| 129 |
if _model_hf2 is None:
|
| 130 |
try:
|
| 131 |
-
|
| 132 |
_model_hf2 = hf_pipeline(
|
| 133 |
"image-classification",
|
| 134 |
model="prithivMLmods/Deep-Fake-Detector-v2-Model",
|
| 135 |
device=hf_device,
|
| 136 |
)
|
| 137 |
-
|
| 138 |
except Exception as exc:
|
| 139 |
logger.warning(f"HF model 2 failed to load: {exc}")
|
| 140 |
_model_hf2 = False
|
|
|
|
| 115 |
|
| 116 |
if _model_hf1 is None:
|
| 117 |
try:
|
| 118 |
+
# HF model 1 loading log suppressed
|
| 119 |
_model_hf1 = hf_pipeline(
|
| 120 |
"image-classification",
|
| 121 |
model="prithivMLmods/Deepfake-Detection-Exp-02-21",
|
| 122 |
device=hf_device,
|
| 123 |
)
|
| 124 |
+
# HF model 1 loaded log suppressed
|
| 125 |
except Exception as exc:
|
| 126 |
logger.warning(f"HF model 1 failed to load: {exc}")
|
| 127 |
_model_hf1 = False
|
| 128 |
|
| 129 |
if _model_hf2 is None:
|
| 130 |
try:
|
| 131 |
+
# HF model 2 loading log suppressed
|
| 132 |
_model_hf2 = hf_pipeline(
|
| 133 |
"image-classification",
|
| 134 |
model="prithivMLmods/Deep-Fake-Detector-v2-Model",
|
| 135 |
device=hf_device,
|
| 136 |
)
|
| 137 |
+
# HF model 2 loaded log suppressed
|
| 138 |
except Exception as exc:
|
| 139 |
logger.warning(f"HF model 2 failed to load: {exc}")
|
| 140 |
_model_hf2 = False
|