Update app.py

app.py

@@ -40,7 +40,7 @@ def get_model_processor():
     model = Qwen2VLForConditionalGeneration.from_pretrained(
         "./showui-2b",
         torch_dtype=torch.bfloat16,
-        device_map="auto",
+        device_map="cuda",
     )
     
     # Load the processor