Spaces:
Running
Running
update reqs.txt, last hope for cudnn. remove return to settings button.
Browse files- app.py +0 -21
- requirements.txt +2 -2
app.py
CHANGED
@@ -502,33 +502,12 @@ def main():
|
|
502 |
st.rerun()
|
503 |
else:
|
504 |
st.error("No output video was generated. Check logs for details.")
|
505 |
-
return_to_input_button = st.button("Return to Settings")
|
506 |
-
if return_to_input_button:
|
507 |
-
st.session_state.page = 'input'
|
508 |
-
st.session_state.temp_dir = None
|
509 |
-
st.session_state.final_video_path = None
|
510 |
-
st.session_state.process_started = False
|
511 |
-
st.rerun()
|
512 |
|
513 |
except subprocess.CalledProcessError as e:
|
514 |
st.error(f"Error running morphing pipeline: {e}")
|
515 |
-
return_to_input_button = st.button("Return to Settings")
|
516 |
-
if return_to_input_button:
|
517 |
-
st.session_state.page = 'input'
|
518 |
-
st.session_state.temp_dir = None
|
519 |
-
st.session_state.final_video_path = None
|
520 |
-
st.session_state.process_started = False
|
521 |
-
st.rerun()
|
522 |
|
523 |
except Exception as e:
|
524 |
st.error(f"An error occurred during processing: {e}")
|
525 |
-
return_to_input_button = st.button("Return to Settings")
|
526 |
-
if return_to_input_button:
|
527 |
-
st.session_state.page = 'input'
|
528 |
-
st.session_state.temp_dir = None
|
529 |
-
st.session_state.final_video_path = None
|
530 |
-
st.session_state.process_started = False
|
531 |
-
st.rerun()
|
532 |
|
533 |
# =============== RESULT PAGE ===============
|
534 |
elif st.session_state.page == 'result':
|
|
|
502 |
st.rerun()
|
503 |
else:
|
504 |
st.error("No output video was generated. Check logs for details.")
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
505 |
|
506 |
except subprocess.CalledProcessError as e:
|
507 |
st.error(f"Error running morphing pipeline: {e}")
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
508 |
|
509 |
except Exception as e:
|
510 |
st.error(f"An error occurred during processing: {e}")
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
511 |
|
512 |
# =============== RESULT PAGE ===============
|
513 |
elif st.session_state.page == 'result':
|
requirements.txt
CHANGED
@@ -14,6 +14,6 @@ torch
|
|
14 |
torchvision
|
15 |
lpips
|
16 |
# peft
|
17 |
-
tensorflow==2.18.0
|
18 |
-
tensorflow_hub==0.16.1
|
19 |
opencv_python
|
|
|
14 |
torchvision
|
15 |
lpips
|
16 |
# peft
|
17 |
+
# tensorflow==2.18.0
|
18 |
+
# tensorflow_hub==0.16.1
|
19 |
opencv_python
|