seifbenayed commited on
Commit
1b403e5
·
1 Parent(s): 1845e2f
Files changed (1) hide show
  1. app.py +4 -4
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
- DEFAULT_DAMAGE_MODEL_PATH = "./model_final.pth" # Replace with your actual path
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: