Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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/
|
| 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/
|
| 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:
|