Update gradio_app.py

gradio_app.py

@@ -240,7 +240,21 @@ def get_image(model, prompts, image1, image2, noise_shape, n_samples=1, ddim_ste
         batch_variants.append(batch_images)
     ## variants, batch, c, t, h, w
     batch_variants = torch.stack(batch_variants)
-    return batch_variants.permute(1, 0, 2, 3, 4, 5)
+    # return batch_variants.permute(1, 0, 2, 3, 4, 5)
+
+    prompt_str = prompt.replace("/", "_slash_") if "/" in prompt else prompt
+    prompt_str = prompt_str.replace(" ", "_") if " " in prompt else prompt_str
+    prompt_str=prompt_str[:40]
+    if len(prompt_str) == 0:
+        prompt_str = 'empty_prompt'
+
+    result_dir = "./tmp/"
+    save_videos(batch_image, result_dir, filenames=[prompt_str], fps=8)
+    print(f"Saved in {prompt_str}. Time used: {(time.time() - start):.2f} seconds")
+    model = model.cpu()
+    saved_result_dir = os.path.join(result_dir, f"{prompt_str}.mp4")
+    print("result saved to:", saved_result_dir)
+    return saved_result_dir