sming256 commited on
Commit
a029659
·
verified ·
1 Parent(s): f94a455

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -0
app.py CHANGED
@@ -200,6 +200,9 @@ class Qwen3VLAutoThinkDemo:
200
  Returns:
201
  Dictionary containing response and metadata
202
  """
 
 
 
203
  # Prepare message
204
  message = [{"role": "system", "content": self.system_prompt}]
205
  content_parts = []
 
200
  Returns:
201
  Dictionary containing response and metadata
202
  """
203
+ if self.model.device.type != "cuda":
204
+ self.model.to("cuda")
205
+
206
  # Prepare message
207
  message = [{"role": "system", "content": self.system_prompt}]
208
  content_parts = []