Update app.py

app.py

@@ -12,6 +12,7 @@ def stream_output(pipe, progress_bar):
     for line in iter(pipe.readline, ''):
         print(line, end='')
         progress_bar.update(1)
+    pipe.close()
 
 HF_TKN = os.environ.get("GATED_HF_TOKEN")
 huggingface_hub.login(token=HF_TKN)
@@ -97,7 +98,7 @@ def infer(ref_video_in, ref_image_in, progress=gr.Progress(track_tqdm=True)):
         noise_aug_strength = 0
         guidance_scale = 2.0
         sample_stride = 2
-        fps = 12
+        fps = 15
         seed = 42
     
         # Create the data structure
@@ -135,7 +136,7 @@ def infer(ref_video_in, ref_image_in, progress=gr.Progress(track_tqdm=True)):
         process = subprocess.Popen(command, stdout=subprocess.PIPE, stderr=subprocess.PIPE, text=True, bufsize=1)
     
         # Create a tqdm progress bar
-        progress_bar = tqdm(total=100, desc="Inference Progress", unit="line", ncols=100)
+        progress_bar = tqdm(total=100, desc="Inference Progress", unit="line", ncols=100, leave=True)
     
         # Create threads to handle stdout and stderr
         stdout_thread = threading.Thread(target=stream_output, args=(process.stdout, progress_bar))
@@ -147,9 +148,9 @@ def infer(ref_video_in, ref_image_in, progress=gr.Progress(track_tqdm=True)):
         stderr_thread.start()
     
         # Wait for the process to complete and the threads to finish
-        process.wait()
         stdout_thread.join()
         stderr_thread.join()
+        process.wait()
 
         # Close the progress bar
         progress_bar.close()