Update project_model.py

project_model.py

@@ -198,8 +198,13 @@ def process_inputs(
 
     vqa_prompt = "You are a helpful visual assistant designed for visually impaired users that assists users by answering their questions. Answer the following question with the help of the shared visual context: " + question + "Shared visual context: " + visual_context 
     
-    # Gemma answer
-    messages = [{
+    # Gemma prompt input
+    messages = [
+        {
+        "role": "assistant",
+        "content": "You are a helpful visual assistant designed for visually impaired users."
+        },
+        {
         "role": "user",
         "content": [
             {"type": "image", "image": session.current_image},
@@ -212,7 +217,7 @@ def process_inputs(
 
     # Handle the output from Gemma model safely
     if isinstance(gemma_output, list) and len(gemma_output) > 0:
-        gemma_text = gemma_output[0]["generated_text"]
+        gemma_text = gemma_output[0]["generated_text"][-1]["content"]
     if isinstance(gemma_text, str):
         answer = gemma_text
     else: