Update app.py

app.py

@@ -63,11 +63,11 @@ args = tyro.cli(ArgumentConfig)
 inference_cfg = partial_fields(InferenceConfig, args.__dict__)  # use attribute of args to initial InferenceConfig
 crop_cfg = partial_fields(CropConfig, args.__dict__)  # use attribute of args to initial CropConfig
 
-gradio_pipeline = GradioPipeline(
-    inference_cfg=inference_cfg,
-    crop_cfg=crop_cfg,
-    args=args
-)
+# gradio_pipeline = GradioPipeline(
+#     inference_cfg=inference_cfg,
+#     crop_cfg=crop_cfg,
+#     args=args
+# )
 
 @spaces.GPU(duration=120)
 def gpu_wrapped_execute_video(*args, **kwargs):
@@ -191,58 +191,77 @@ def run_end_to_end(image_path, text, voice, input_video, flag_relative, flag_do_
 
 
 
+
+###### 테스트중 ######
+    
+
+stf_pipeline = STFPipeline()
+
 with gr.Blocks(theme=gr.themes.Soft()) as demo:
-    with gr.Tabs():
-        # 첫 번째 탭: Text to LipSync
-        with gr.Tab("Text to LipSync"):
-            gr.Markdown("# Text to LipSync")
-            with gr.Row():
-                script_txt = gr.Text()
-                voice = gr.Audio(label="사용자 음성", type="filepath")
-                input_video = gr.Video()
-
-            with gr.Row():
-                image_input = gr.Image(type="filepath")  # 여기서 image_input을 정의합니다.
-                output_video.render()
-                #crop_output_video.render()
-                output_video_concat.render()
-
-            with gr.Row():
-                flag_relative_input = gr.Checkbox(value=True, label="relative motion")
-                flag_do_crop_input = gr.Checkbox(value=True, label="do crop")
-                flag_remap_input = gr.Checkbox(value=True, label="paste-back")
-                flag_crop_driving_video_input = gr.Checkbox(value=False, label="do crop (driving video)")
-                male = gr.Checkbox(value=False, label="male")
-                #animal = gr.Checkbox(value=False, label="animal")  # animal 체크박스 추가
-
-            with gr.Row():
-                generate_speech = gr.Button("🚀 Generate Speech", variant="primary")
-
-            generate_speech.click(
-                fn=run_end_to_end,
+    stf_button = gr.Button("stf test", variant="primary")
+    stf_button.click(
+                fn=gpu_wrapped_stf_pipeline_execute,
                 inputs=[
-                    image_input,
-                    script_txt,
-                    voice,
-                    input_video,
-                    flag_relative_input,
-                    flag_do_crop_input,
-                    flag_remap_input,
-                    flag_crop_driving_video_input,
-                    male,
-                    #animal  # 추가된 animal 입력
+                    audio_path
                 ],
-                outputs=[output_video, output_video_concat]
+                outputs=[driving_video_path]
             )
 
-        # # 두 번째 탭: FLUX 이미지 생성
-        # with gr.Tab("FLUX 이미지 생성"):
-        #     flux_tab(image_input)  # FLUX 이미지 생성을 위한 별도의 탭
+###### 테스트중 ######
+
+
+# with gr.Blocks(theme=gr.themes.Soft()) as demo:
+#     with gr.Tabs():
+#         # 첫 번째 탭: Text to LipSync
+#         with gr.Tab("Text to LipSync"):
+#             gr.Markdown("# Text to LipSync")
+#             with gr.Row():
+#                 script_txt = gr.Text()
+#                 voice = gr.Audio(label="사용자 음성", type="filepath")
+#                 input_video = gr.Video()
+
+#             with gr.Row():
+#                 image_input = gr.Image(type="filepath")  # 여기서 image_input을 정의합니다.
+#                 output_video.render()
+#                 #crop_output_video.render()
+#                 output_video_concat.render()
+
+#             with gr.Row():
+#                 flag_relative_input = gr.Checkbox(value=True, label="relative motion")
+#                 flag_do_crop_input = gr.Checkbox(value=True, label="do crop")
+#                 flag_remap_input = gr.Checkbox(value=True, label="paste-back")
+#                 flag_crop_driving_video_input = gr.Checkbox(value=False, label="do crop (driving video)")
+#                 male = gr.Checkbox(value=False, label="male")
+#                 #animal = gr.Checkbox(value=False, label="animal")  # animal 체크박스 추가
+
+#             with gr.Row():
+#                 generate_speech = gr.Button("🚀 Generate Speech", variant="primary")
+
+#             generate_speech.click(
+#                 fn=run_end_to_end,
+#                 inputs=[
+#                     image_input,
+#                     script_txt,
+#                     voice,
+#                     input_video,
+#                     flag_relative_input,
+#                     flag_do_crop_input,
+#                     flag_remap_input,
+#                     flag_crop_driving_video_input,
+#                     male,
+#                     #animal  # 추가된 animal 입력
+#                 ],
+#                 outputs=[output_video, output_video_concat]
+#             )
+
+#         # # 두 번째 탭: FLUX 이미지 생성
+#         # with gr.Tab("FLUX 이미지 생성"):
+#         #     flux_tab(image_input)  # FLUX 이미지 생성을 위한 별도의 탭
 
-        # # 세 번째 탭: Flux 개발용 탭
-        # with gr.Tab("FLUX Dev"):
-        #     flux_demo = create_flux_tab()  # Flux 개발용 탭 생성
-        #     #flux_demo.render()  # 해당 UI를 별도의 탭에만 렌더링
+#         # # 세 번째 탭: Flux 개발용 탭
+#         # with gr.Tab("FLUX Dev"):
+#         #     flux_demo = create_flux_tab()  # Flux 개발용 탭 생성
+#         #     #flux_demo.render()  # 해당 UI를 별도의 탭에만 렌더링