seifbenayed commited on
Commit
230092e
·
verified ·
1 Parent(s): e2914ad

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -172,7 +172,7 @@ def process_image(input_image, damage_model_path, deepfake_model_path, deepfake_
172
  # Default model paths if not provided
173
  damage_model_path = damage_model_path or "./model_final.pth"
174
  deepfake_model_path = deepfake_model_path or "./PoseEfficientNet_custom_laanet_model_final.pth"
175
- deepfake_cfg_path = deepfake_cfg_path or "./configs/detector/detector2.yaml"
176
 
177
  # Check if we're running in demo mode (no real models available)
178
  models_exist, model_status = check_model_files(damage_model_path, deepfake_model_path, deepfake_cfg_path)
@@ -233,7 +233,7 @@ def create_gradio_interface():
233
  # Check default model paths
234
  default_damage_path = "./model_final.pth"
235
  default_deepfake_path = "./PoseEfficientNet_custom_laanet_model_final.pth"
236
- default_config_path = "./configs/detector/detector2.yaml"
237
 
238
  # Make sure we have a safe version of check_model_files
239
  try:
 
172
  # Default model paths if not provided
173
  damage_model_path = damage_model_path or "./model_final.pth"
174
  deepfake_model_path = deepfake_model_path or "./PoseEfficientNet_custom_laanet_model_final.pth"
175
+ deepfake_cfg_path = deepfake_cfg_path or "./configs/detector2.yaml"
176
 
177
  # Check if we're running in demo mode (no real models available)
178
  models_exist, model_status = check_model_files(damage_model_path, deepfake_model_path, deepfake_cfg_path)
 
233
  # Check default model paths
234
  default_damage_path = "./model_final.pth"
235
  default_deepfake_path = "./PoseEfficientNet_custom_laanet_model_final.pth"
236
+ default_config_path = "./configs/detector2.yaml"
237
 
238
  # Make sure we have a safe version of check_model_files
239
  try: