saa231 commited on
Commit
52613c9
·
verified ·
1 Parent(s): 58d9eeb

Update project_model.py

Browse files
Files changed (1) hide show
  1. project_model.py +1 -1
project_model.py CHANGED
@@ -211,7 +211,7 @@ def process_inputs(
211
  gemma_input = [{
212
  "role": "user",
213
  "content": [
214
- {"type": "images", "images": session.annotated_image}, # Image context
215
  {"type": "text", "text": question + " " + session.visual_context} # Question and visual context combined as text
216
  ]
217
  }]
 
211
  gemma_input = [{
212
  "role": "user",
213
  "content": [
214
+ {"type": "images", "images": session.current_image}, # Image context
215
  {"type": "text", "text": question + " " + session.visual_context} # Question and visual context combined as text
216
  ]
217
  }]