Stremly commited on
Commit
3069723
Β·
verified Β·
1 Parent(s): 49fd159

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +12 -2
app.py CHANGED
@@ -67,11 +67,21 @@ def navigate(screenshot, task: str, platform: str, history):
67
  print(messages)
68
 
69
  # ─────────────────────────── model forward ─────────────────────────────
 
 
 
 
 
 
 
 
 
 
70
  text = processor.apply_chat_template(
71
  messages, tokenize=False, add_generation_prompt=True
72
  )
73
- images, videos = process_vision_info(messages)
74
- print(images)
75
  inputs = processor(
76
  text=[text],
77
  images=images,
 
67
  print(messages)
68
 
69
  # ─────────────────────────── model forward ─────────────────────────────
70
+
71
+ images, videos = process_vision_info(messages)
72
+ i=0
73
+ for message in messages:
74
+ if message['role'] == 'user' and isinstance(message.get('content'), list):
75
+ for item in message['content']:
76
+ if item.get('type') == 'image_url' and isinstance(item.get('image_url'), str):
77
+ item['image_url'] = images[0]
78
+ i+=1
79
+
80
  text = processor.apply_chat_template(
81
  messages, tokenize=False, add_generation_prompt=True
82
  )
83
+ print("\nimages\n:",images)
84
+ print("\ntext\n",text)
85
  inputs = processor(
86
  text=[text],
87
  images=images,