saa231 commited on
Commit
acdebd9
·
verified ·
1 Parent(s): b72cd3e

Update project_model.py

Browse files
Files changed (1) hide show
  1. project_model.py +1 -1
project_model.py CHANGED
@@ -204,7 +204,7 @@ def process_inputs(
204
 
205
  # Create the structured input for Gemma, including image, question, and visual context
206
  gemma_input = {
207
- "image": session.annotated_image, # Image context
208
  "text": question + " " + session.visual_context # Question and visual context combined as text
209
  }
210
 
 
204
 
205
  # Create the structured input for Gemma, including image, question, and visual context
206
  gemma_input = {
207
+ "images": session.annotated_image, # Image context
208
  "text": question + " " + session.visual_context # Question and visual context combined as text
209
  }
210