Update app.py

app.py

@@ -52,13 +52,29 @@ print("LD_LIBRARY_PATH:", os.environ['LD_LIBRARY_PATH'])
 from stf_utils import STFPipeline
 
 stf_pipeline = STFPipeline()
+audio_path="assets/examples/driving/test_aud.mp3"
 
 @spaces.GPU(duration=120)
 def gpu_wrapped_stf_pipeline_execute(audio_path):
     return stf_pipeline.execute(audio_path)
 
     
+###### 테스트중 ######
+    
+
+stf_pipeline = STFPipeline()
 
+with gr.Blocks(theme=gr.themes.Soft()) as demo:
+    stf_button = gr.Button("stf test", variant="primary")
+    stf_button.click(
+                fn=gpu_wrapped_stf_pipeline_execute,
+                inputs=[
+                    audio_path
+                ],
+                outputs=[driving_video_path]
+            )
+
+###### 테스트중 ######