V0.1

app.py

@@ -520,6 +520,26 @@ def auto_install_dependencies():
         print(f"Error installing dependencies: {e}")
         return False
 
+def check_model_paths(damage_path, deepfake_path):
+    """Check if model paths are valid and exist"""
+    output = ["## Path Verification Results\n"]
+    
+    # Check damage model
+    if os.path.exists(damage_path):
+        file_size = os.path.getsize(damage_path) / (1024 * 1024)  # Size in MB
+        output.append(f"✅ **Damage model:** Found at {damage_path} ({file_size:.2f} MB)")
+    else:
+        output.append(f"❌ **Damage model:** NOT found at {damage_path}")
+    
+    # Check deepfake model
+    if os.path.exists(deepfake_path):
+        file_size = os.path.getsize(deepfake_path) / (1024 * 1024)  # Size in MB
+        output.append(f"✅ **ViT deepfake model:** Found at {deepfake_path} ({file_size:.2f} MB)")
+    else:
+        output.append(f"❌ **ViT deepfake model:** NOT found at {deepfake_path}")
+    
+    return "\n".join(output)
+
 def create_gradio_interface():
     # Define a theme
     theme = gr.themes.Soft(