Update opensora/serve/gradio_web_server.py

opensora/serve/gradio_web_server.py

@@ -24,9 +24,9 @@ from opensora.models.diffusion.latte.modeling_latte import LatteT2V
 from opensora.sample.pipeline_videogen import VideoGenPipeline
 from opensora.serve.gradio_utils import block_css, title_markdown, randomize_seed_fn, set_env, examples, DESCRIPTION
 
-#import spaces
+import spaces
 
-#@spaces.GPU
+@spaces.GPU
 def generate_img(prompt, sample_steps, scale, seed=0, randomize_seed=False, force_images=False):
     seed = int(randomize_seed_fn(seed, randomize_seed))
     set_env(seed)