mrneuralnet commited on
Commit
db1fe34
·
1 Parent(s): 1f066c6

Add setup function

Browse files
Files changed (2) hide show
  1. .gitignore +3 -1
  2. app.py +6 -0
.gitignore CHANGED
@@ -1,2 +1,4 @@
1
  temps/*
2
- !temps/.gitkeep
 
 
 
1
  temps/*
2
+ !temps/.gitkeep
3
+
4
+ env
app.py CHANGED
@@ -15,6 +15,8 @@ from PIL import Image
15
  import shlex
16
  import subprocess
17
 
 
 
18
  def main():
19
  st.markdown("###")
20
  uploaded_file = st.file_uploader('Upload a picture', type=['mp4', 'jpg', 'jpeg', 'png'], accept_multiple_files=False)
@@ -63,10 +65,14 @@ def main():
63
  st.text(pred)
64
 
65
 
 
 
 
66
 
67
  if __name__ == "__main__":
68
  st.set_page_config(
69
  page_title="Nodeflux Deepfake Detection", page_icon=":pencil2:"
70
  )
71
  st.title("Deepfake Detection")
 
72
  main()
 
15
  import shlex
16
  import subprocess
17
 
18
+ import os
19
+
20
  def main():
21
  st.markdown("###")
22
  uploaded_file = st.file_uploader('Upload a picture', type=['mp4', 'jpg', 'jpeg', 'png'], accept_multiple_files=False)
 
65
  st.text(pred)
66
 
67
 
68
+ def setup():
69
+ if not os.path.isdir("temps"):
70
+ os.makedirs("temps")
71
 
72
  if __name__ == "__main__":
73
  st.set_page_config(
74
  page_title="Nodeflux Deepfake Detection", page_icon=":pencil2:"
75
  )
76
  st.title("Deepfake Detection")
77
+ setup()
78
  main()