Spaces:
Sleeping
Sleeping
Commit
·
db1fe34
1
Parent(s):
1f066c6
Add setup function
Browse files- .gitignore +3 -1
- 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()
|