AdeelHassan commited on
Commit
3c3947d
·
verified ·
1 Parent(s): 7be4b16
Files changed (1) hide show
  1. 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
- logger.info("Loading HF model 1: Deepfake-Detection-Exp-02-21 ...")
119
  _model_hf1 = hf_pipeline(
120
  "image-classification",
121
  model="prithivMLmods/Deepfake-Detection-Exp-02-21",
122
  device=hf_device,
123
  )
124
- logger.debug("HF model 1 loaded.")
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
- logger.debug("Loading HF model 2: Deep-Fake-Detector-v2-Model ...")
132
  _model_hf2 = hf_pipeline(
133
  "image-classification",
134
  model="prithivMLmods/Deep-Fake-Detector-v2-Model",
135
  device=hf_device,
136
  )
137
- logger.debug("HF model 2 loaded.")
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