Fabrice-TIERCELIN commited on
Commit
632c988
1 Parent(s): bf92825

Space annotation

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -1,3 +1,4 @@
 
1
  import gradio as gr
2
  import numpy as np
3
  from PIL import Image
@@ -48,6 +49,7 @@ def convert_mp4_to_frames(video_path, duration=3):
48
 
49
  return frames
50
 
 
51
  def infer(prompt, video_in, denoise_strength):
52
 
53
  negative_prompt = "text, watermark, copyright, blurry, nsfw"
 
1
+ import spaces
2
  import gradio as gr
3
  import numpy as np
4
  from PIL import Image
 
49
 
50
  return frames
51
 
52
+ @spaces.GPU(duration=120)
53
  def infer(prompt, video_in, denoise_strength):
54
 
55
  negative_prompt = "text, watermark, copyright, blurry, nsfw"