Spaces:
Runtime error
Runtime error
Commit
·
1b403e5
1
Parent(s):
1845e2f
V1.1
Browse files
app.py
CHANGED
|
@@ -26,7 +26,9 @@ import gradio as gr
|
|
| 26 |
from PIL import Image
|
| 27 |
from torchvision import transforms
|
| 28 |
|
| 29 |
-
|
|
|
|
|
|
|
| 30 |
|
| 31 |
# Add current directory to path
|
| 32 |
if not os.getcwd() in sys.path:
|
|
@@ -507,9 +509,7 @@ def process_image(input_image, damage_model_path, deepfake_model_path, deepfake_
|
|
| 507 |
return result_img, "\n".join(progress_info)
|
| 508 |
|
| 509 |
|
| 510 |
-
|
| 511 |
-
DEFAULT_DEEPFAKE_MODEL_PATH = "./PoseEfficientNet_custom_laanet_model_final.pth" # Replace with your actual path
|
| 512 |
-
DEFAULT_DEEPFAKE_CFG_PATH = "./configs/detector2.yaml" # Replace with your actual path
|
| 513 |
|
| 514 |
def create_gradio_interface():
|
| 515 |
with gr.Blocks(title="Car Damage & Deepfake Detection") as app:
|
|
|
|
| 26 |
from PIL import Image
|
| 27 |
from torchvision import transforms
|
| 28 |
|
| 29 |
+
DEFAULT_DAMAGE_MODEL_PATH = "./model_final.pth" # Replace with your actual path
|
| 30 |
+
DEFAULT_DEEPFAKE_MODEL_PATH = "./PoseEfficientNet_custom_laanet_model_final.pth" # Replace with your actual path
|
| 31 |
+
DEFAULT_DEEPFAKE_CFG_PATH = "./configs/detector2.yaml" # Replace with your actual path
|
| 32 |
|
| 33 |
# Add current directory to path
|
| 34 |
if not os.getcwd() in sys.path:
|
|
|
|
| 509 |
return result_img, "\n".join(progress_info)
|
| 510 |
|
| 511 |
|
| 512 |
+
|
|
|
|
|
|
|
| 513 |
|
| 514 |
def create_gradio_interface():
|
| 515 |
with gr.Blocks(title="Car Damage & Deepfake Detection") as app:
|