Update project_model.py

project_model.py

@@ -210,8 +210,7 @@ def process_inputs(
 
     # Make sure the output is in the expected format (a string)
     if isinstance(gemma_output, list) and len(gemma_output) > 0:
-        # Assuming it's a list of dicts with the 'generated_text' field
-        gemma_text = gemma_output[0].get("generated_text", "")
+        gemma_text = gemma_output[0]["generated_text"][-1]["content"]
         if isinstance(gemma_text, str):
             answer = gemma_text
         else: