Update app.py

app.py

@@ -4,12 +4,12 @@ from transformers import Qwen2_5_VLForConditionalGeneration, AutoProcessor
 from qwen_vl_utils import process_vision_info
 from PIL import Image
 import torch
-import os
+import os, time
 
 # Load the model and processor
 model = Qwen2_5_VLForConditionalGeneration.from_pretrained(
     "daniel3303/QwenStoryteller",
-    torch_dtype="auto",
+    torch_dtype=torch.float16,
     device_map="auto"
 )
 processor = AutoProcessor.from_pretrained("daniel3303/QwenStoryteller")
@@ -87,4 +87,4 @@ with gr.Blocks() as demo:
     gen_button.click(generate_story, upload_button, outputs)
 
 if __name__ == "__main__":
-    demo.launch()
+    demo.queue().launch(show_errors=True)