Spaces:

weizmannscience
/

tokenflow

Running on Zero

multimodalart HF Staff commited on Sep 19

Commit

ba1c8f4

verified ·

1 Parent(s): 1e8a534

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -110,7 +110,7 @@ def prep(config):
         frames, latents, total_inverted_latents = frames_and_latents
         return frames, latents, total_inverted_latents, rgb_reconstruction
 def preprocess_and_invert(input_video,
                           frames,
                           latents,
@@ -166,7 +166,7 @@ def preprocess_and_invert(input_video,
     return frames, latents, inverted_latents, zs, do_inversion
 def edit_with_pnp(input_video,
                   frames,
                   latents,

         frames, latents, total_inverted_latents = frames_and_latents
         return frames, latents, total_inverted_latents, rgb_reconstruction
+@spaces.GPU(duration=120)
 def preprocess_and_invert(input_video,
                           frames,
                           latents,
     return frames, latents, inverted_latents, zs, do_inversion
+@spaces.GPU(duration=120)
 def edit_with_pnp(input_video,
                   frames,
                   latents,