ai-tube-model-als-9

Sleeping

wangfuyun commited on Feb 26

Commit

b470517

•

1 Parent(s): 8b77dff

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -1,5 +1,5 @@
 import gradio as gr
 # import gradio.helpers
 import torch
 import os
@@ -61,7 +61,7 @@ model_select("AnimateLCM-SVD-xt.safetensors")
 max_64_bit_int = 2**63 - 1
 def sample(
     image: Image,
     seed: Optional[int] = 42,

 import gradio as gr
+import spaces
 # import gradio.helpers
 import torch
 import os
 max_64_bit_int = 2**63 - 1
+@spaces.GPU
 def sample(
     image: Image,
     seed: Optional[int] = 42,