nalin0503 commited on
Commit
a3f83c0
·
1 Parent(s): d9f185c

update reqs.txt, last hope for cudnn. remove return to settings button.

Browse files
Files changed (2) hide show
  1. app.py +0 -21
  2. 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