Update app.py

app.py

@@ -158,9 +158,9 @@ def process_image(input_image, damage_model_path, deepfake_model_path, deepfake_
         return demo_mode_process(input_image)
     
     # Default model paths if not provided
-    damage_model_path = damage_model_path or "./training/output/model_final.pth"
-    deepfake_model_path = deepfake_model_path or "./logs/13-04-2025/PoseEfficientNet_custom_laanet_model_final.pth"
-    deepfake_cfg_path = deepfake_cfg_path or "./training/configs/detector/detector2.yaml"
+    damage_model_path = damage_model_path or "./model_final.pth"
+    deepfake_model_path = deepfake_model_path or "./PoseEfficientNet_custom_laanet_model_final.pth"
+    deepfake_cfg_path = deepfake_cfg_path or "./configs/detector/detector2.yaml"
     
     # Check if we're running in demo mode (no real models available)
     models_exist, model_status = check_model_files(damage_model_path, deepfake_model_path, deepfake_cfg_path)
@@ -219,9 +219,9 @@ def create_gradio_interface():
         status_message += "❌ Custom models module import failed - check your installation\n"
         
     # Check default model paths
-    default_damage_path = "./training/output/model_final.pth"
-    default_deepfake_path = "./logs/13-04-2025/PoseEfficientNet_custom_laanet_model_final.pth"
-    default_config_path = "./training/configs/detector/detector2.yaml"
+    default_damage_path = "./output/model_final.pth"
+    default_deepfake_path = "./PoseEfficientNet_custom_laanet_model_final.pth"
+    default_config_path = "./configs/detector/detector2.yaml"
     
     # Make sure we have a safe version of check_model_files
     try:
@@ -234,7 +234,7 @@ def create_gradio_interface():
         model_status = []
     
     # Check if example images exist
-    example_images = ["examples/car_damage1.jpg", "examples/car_damage2.jpg"]
+    example_images = ["./test5.png", "./test3.png"]
     valid_examples = []
     example_status = []