wangfuyun commited on
Commit
b470517
1 Parent(s): 8b77dff

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -1,5 +1,5 @@
1
  import gradio as gr
2
-
3
  # import gradio.helpers
4
  import torch
5
  import os
@@ -61,7 +61,7 @@ model_select("AnimateLCM-SVD-xt.safetensors")
61
 
62
  max_64_bit_int = 2**63 - 1
63
 
64
-
65
  def sample(
66
  image: Image,
67
  seed: Optional[int] = 42,
 
1
  import gradio as gr
2
+ import spaces
3
  # import gradio.helpers
4
  import torch
5
  import os
 
61
 
62
  max_64_bit_int = 2**63 - 1
63
 
64
+ @spaces.GPU
65
  def sample(
66
  image: Image,
67
  seed: Optional[int] = 42,