Stremly commited on
Commit
27ea670
·
verified ·
1 Parent(s): 454ec10

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -16,7 +16,7 @@ model = Qwen2_5_VLForConditionalGeneration.from_pretrained(
16
  )
17
  processor = AutoProcessor.from_pretrained(
18
  "ByteDance-Seed/UI-TARS-1.5-7B",
19
- size={"shortest_edge": 256 * 28 * 28, "longest_edge": 1344 * 28 * 28},
20
  use_fast=True,
21
  )
22
 
@@ -63,7 +63,7 @@ def navigate(screenshot, task: str, platform: str, history):
63
  "## Action Space\nclick(start_box='...') / type(...)\n\n"
64
  f"### Task\n{task}"
65
  )
66
- user_history = [{"role":"user","content":prompt_header},{"role":"user","content":{"type": "image_url", "image_url":screenshot}}]
67
  messages = user_history
68
 
69
 
 
16
  )
17
  processor = AutoProcessor.from_pretrained(
18
  "ByteDance-Seed/UI-TARS-1.5-7B",
19
+ size={"shortest_edge": 100 * 28 * 28, "longest_edge": 16384 * 28 * 28},
20
  use_fast=True,
21
  )
22
 
 
63
  "## Action Space\nclick(start_box='...') / type(...)\n\n"
64
  f"### Task\n{task}"
65
  )
66
+ user_history = [{"role":"user","content":[{"type":"text","text":prompt_header},{"type": "image_url", "image_url":screenshot}]}
67
  messages = user_history
68
 
69