Update app.py

app.py

@@ -34,9 +34,7 @@ source_file = st.file_uploader("Upload source image")
 target_file = st.file_uploader("Upload target image/video")
 use_face_enhancer = st.checkbox("Use only Face Enhancer", False)
 
-if st.button("Swap Faces"):
+if source_file and target_file and st.button("Swap Faces"):
     result = run_scripts(target_file.name, source_file.name, use_face_enhancer)
     if result:
         st.image(result)  # Display the result as an image
-
-# No need to remove uploaded files, as they are now passed as strings.