Update inference/flovd_demo.py

inference/flovd_demo.py

@@ -396,6 +396,7 @@ def generate_video(
     """
         # Inference time
     """
+    print("Inference time from image to video.")
     image = rearrange((torch.tensor(np.array(image)).to(torch.float) / 255. * 2. - 1.).unsqueeze(0), 'b h w c -> b c h w')
     image = image.to(local_rank)
     prompt_short = prompt[:30].strip().replace(" ", "_")
@@ -452,7 +453,9 @@ def generate_video(
             ).frames[0]
         
     # Save Result
+    print("video content : ", video_generate)
     save_path = os.path.join(output_video_path, f"{prompt_short}_{cam_name}.mp4")
+    print("video generated successfully")
     export_to_video(video_generate, save_path, fps=fps)