saa231 commited on
Commit
430f05a
·
verified ·
1 Parent(s): 5a72575

Update project_model.py

Browse files
Files changed (1) hide show
  1. project_model.py +1 -1
project_model.py CHANGED
@@ -197,7 +197,7 @@ def process_inputs(
197
  # Send improved prompt
198
  gemma_output = gemma_pipe(text=full_prompt, max_new_tokens=200)
199
 
200
- answer = gemma_output[0]["generated_text"][-1]["content"]
201
 
202
  session.add_answer(answer)
203
 
 
197
  # Send improved prompt
198
  gemma_output = gemma_pipe(text=full_prompt, max_new_tokens=200)
199
 
200
+ answer = gemma_output
201
 
202
  session.add_answer(answer)
203