Update app.py

app.py

@@ -93,7 +93,51 @@ examples = [
 ]
 
 def generate_examples(input_image, prompt):
-    yield process(input_image, prompt)
+    
+    t2v=False 
+    n_prompt=""
+    seed=31337
+    total_second_length=5 
+    latent_window_size=9 
+    steps=25 
+    cfg=1.0 
+    gs=10.0 
+    rs=0.0
+    gpu_memory_preservation=6 
+    use_teacache=True 
+    mp4_crf=16
+
+    global stream
+    
+    # assert input_image is not None, 'No input image!'
+    if t2v:
+        default_height, default_width = 640, 640
+        input_image = np.ones((default_height, default_width, 3), dtype=np.uint8) * 255
+        print("No input image provided. Using a blank white image.") 
+
+    yield None, None, '', '', gr.update(interactive=False), gr.update(interactive=True)
+
+    stream = AsyncStream()
+
+    async_run(worker, input_image, prompt, n_prompt, seed, total_second_length, latent_window_size, steps, cfg, gs, rs, gpu_memory_preservation, use_teacache, mp4_crf)
+
+    output_filename = None
+
+    while True:
+        flag, data = stream.output_queue.next()
+
+        if flag == 'file':
+            output_filename = data
+            yield output_filename, gr.update(), gr.update(), gr.update(), gr.update(interactive=False), gr.update(interactive=True)
+
+        if flag == 'progress':
+            preview, desc, html = data
+            yield gr.update(), gr.update(visible=True, value=preview), desc, html, gr.update(interactive=False), gr.update(interactive=True)
+
+        if flag == 'end':
+            yield output_filename, gr.update(visible=False), gr.update(), '', gr.update(interactive=True), gr.update(interactive=False)
+            break
+
 
     
 @torch.no_grad()