Update app.py

app.py

@@ -1,3 +1,4 @@
+import spaces
 import gradio as gr
 from transformers import Qwen2_5_VLForConditionalGeneration, AutoProcessor
 from qwen_vl_utils import process_vision_info
@@ -12,6 +13,7 @@ model = Qwen2_5_VLForConditionalGeneration.from_pretrained(
 )
 processor = AutoProcessor.from_pretrained("daniel3303/QwenStoryteller")
 
+@spaces.GPU()
 def generate_story(images):
     image_content = []
     for img in images[:6]: