From 85d732083749e1f99df3a9ff82c4d2827afa4e81 Mon Sep 17 00:00:00 2001
From: Chris Malone <chrismrulz@hotmail.com>
Date: Wed, 26 Mar 2025 18:05:07 +1100
Subject: [PATCH] re-removed yield statements that prevented it running

---
 gradio_server.py | 8 +-------
 1 file changed, 1 insertion(+), 7 deletions(-)

diff --git a/gradio_server.py b/gradio_server.py
index f38169c..b9a23c4 100644
--- a/gradio_server.py
+++ b/gradio_server.py
@@ -1431,7 +1431,7 @@ def generate_video(
             end_time = time.time()
             abort = True
             state["prompt"] = ""
-            yield f"Video generation was aborted. Total Generation Time: {end_time-start_time:.1f}s"
+            print(f"Video generation was aborted. Total Generation Time: {end_time-start_time:.1f}s")
         else:
             sample = samples.cpu()
             # video = rearrange(sample.cpu().numpy(), "c t h w -> t h w c")
@@ -1465,12 +1465,6 @@ def generate_video(
 
             print(f"New video saved to Path: "+video_path)
             file_list.append(video_path)
-            if video_no < total_video:
-                yield  status 
-            else:
-                end_time = time.time()
-                state["prompt"] = ""
-                yield f"Total Generation Time: {end_time-start_time:.1f}s"
         seed += 1
         repeat_no += 1