Update project_model.py

project_model.py

@@ -208,21 +208,24 @@ def process_inputs(
     #    "text": question + " " + session.visual_context  # Question and visual context combined as text
     #}
 
-    gemma_input = {
-        "images": session.current_image,   # Image context
-         "text": question + " This is the shared visual context: " + session.visual_context  # Question and visual context combined as text
-        }
+    #gemma_input = {
+     #   "images": session.current_image,   # Image context
+      #   "text": question + " This is the shared visual context: " + session.visual_context  # Question and visual context combined as text
+       # }
     
     # Process the input through Gemma
-    gemma_output = gemma_pipe(gemma_input)
+    gemma_output = gemma_pipe(
+        images=session.current_image,
+        text=question + " This is the shared visual context: " + session.visual_context
+        )
 
     # Handle the output from Gemma model safely
     if isinstance(gemma_output, list) and len(gemma_output) > 0:
-        gemma_text = gemma_output[0]["generated_text"][-1]["content"]
-        if isinstance(gemma_text, str):
-            answer = gemma_text
-        else:
-            answer = "No valid generated text found."
+        gemma_text = gemma_output[0]["generated_text"]
+    if isinstance(gemma_text, str):
+        answer = gemma_text
+    else:
+        answer = "No valid generated text found."
     else:
         answer = "No valid output from Gemma model."