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()
|