Update opensora/serve/gradio_web_server.py

opensora/serve/gradio_web_server.py

@@ -26,13 +26,13 @@ from opensora.serve.gradio_utils import block_css, title_markdown, randomize_see
 from opensora.serve.gradio_utils import examples_txt, examples
 
 @spaces.GPU(duration=300)
-@torch.inference_mode()
-
 @torch.inference_mode()
 def generate_img(prompt, sample_steps, scale, seed=0, randomize_seed=False, force_images=False):
     video_length = transformer_model.config.video_length if not force_images else 1
     height, width = int(args.version.split('x')[1]), int(args.version.split('x')[2])
     num_frames = 1 if video_length == 1 else int(args.version.split('x')[0])
+    print(prompt in examples_txt)
+    print(prompt, examples_txt)
     if not force_images and prompt in examples_txt:
         idx = examples_txt.index(prompt)
         tmp_save_path = f'demo65-221/f65/{idx+1}.mp4'