Spaces:
azhan77168
/
Running on Zero

azhan77168 commited on
Commit
a7fa216
·
verified ·
1 Parent(s): 237cc11

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -1,3 +1,4 @@
 
1
  import copy
2
  import os
3
  import random
@@ -53,7 +54,7 @@ def t2v_generation(prompt, resolution, watermark_wan, seed=-1):
53
  gr.Warning(f"Warning: {e}")
54
  return None, gr.Button(visible=True)
55
 
56
-
57
  def t2v_generation_async(prompt, size, watermark_wan, seed=-1):
58
  print(seed)
59
  seed = seed if seed >= 0 else random.randint(0, 2147483647)
 
1
+ import spaces
2
  import copy
3
  import os
4
  import random
 
54
  gr.Warning(f"Warning: {e}")
55
  return None, gr.Button(visible=True)
56
 
57
+ @spaces.GPU
58
  def t2v_generation_async(prompt, size, watermark_wan, seed=-1):
59
  print(seed)
60
  seed = seed if seed >= 0 else random.randint(0, 2147483647)