seifbenayed commited on
Commit
fdd9119
·
1 Parent(s): 86e5923
Files changed (1) hide show
  1. app.py +7 -0
app.py CHANGED
@@ -26,6 +26,13 @@ import gradio as gr
26
  from PIL import Image
27
  from torchvision import transforms
28
 
 
 
 
 
 
 
 
29
  # Add current directory to path
30
  if not os.getcwd() in sys.path:
31
  sys.path.append(os.getcwd())
 
26
  from PIL import Image
27
  from torchvision import transforms
28
 
29
+
30
+ damage_model_path = "./output/model_final.pth"
31
+ deepfake_model_path = "./output/PoseEfficientNet_custom_laanet_model_final.pth"
32
+ deepfake_cfg_path = "./configs/detector2.yaml"
33
+
34
+
35
+
36
  # Add current directory to path
37
  if not os.getcwd() in sys.path:
38
  sys.path.append(os.getcwd())