predict now yields

zsxkib · Sep 18, 2023 · 8ef3f1b · 8ef3f1b
1 parent cb87929
commit 8ef3f1b
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/predict.py b/predict.py
@@ -2,7 +2,7 @@
 # https://github.com/replicate/cog/blob/main/docs/python.md
 
 from types import SimpleNamespace
-from typing import List
+from typing import List, Generator
 import subprocess
 import torch
 import torchvision.transforms.functional as TF
@@ -655,7 +655,7 @@ def predict(
             subprocess.run(cmd)
 
             # Append the output path and prepare for the next iteration if needed
-            output_paths.append(Path(mp4_outname))
             mp4 = mp4_outname
+            yield Path(mp4_outname)
 
         return output_paths