1inkusFace commited on
Commit
cca49ed
·
verified ·
1 Parent(s): 5af50c2

Update app.py

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 argparse
3
  import sys
@@ -28,6 +29,7 @@ def init_predictor():
28
  )
29
  )
30
 
 
31
  def generate_video(prompt, seed, image=None):
32
  print(f"image:{type(image)}")
33
  if seed == -1:
 
1
+ import spaces
2
  import gradio as gr
3
  import argparse
4
  import sys
 
29
  )
30
  )
31
 
32
+ @spaces.GPU(duration=80)
33
  def generate_video(prompt, seed, image=None):
34
  print(f"image:{type(image)}")
35
  if seed == -1: