Bhanu-Chander-ABB commited on
Commit
f7728a8
·
1 Parent(s): 7db1ce5

modified final answer condition

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -914,9 +914,9 @@ def run_and_submit_all( profile: gr.OAuthProfile | None):
914
  # submitted_answer = agent.run(full_prompt)
915
  # submitted_answer_raw = agent.run(question_text)
916
  submitted_answer = agent.run(question_text)
917
- try:
918
  scraped_answer = submitted_answer.split("YOUR FINAL ANSWER:")[-1].strip()
919
- except:
920
  scraped_answer = submitted_answer
921
  # submitted_answer = extract_final_answer(submitted_answer_raw)
922
  answers_payload.append({"task_id": task_id, "submitted_answer": scraped_answer})
 
914
  # submitted_answer = agent.run(full_prompt)
915
  # submitted_answer_raw = agent.run(question_text)
916
  submitted_answer = agent.run(question_text)
917
+ if "YOUR FINAL ANSWER:" in submitted_answer:
918
  scraped_answer = submitted_answer.split("YOUR FINAL ANSWER:")[-1].strip()
919
+ else:
920
  scraped_answer = submitted_answer
921
  # submitted_answer = extract_final_answer(submitted_answer_raw)
922
  answers_payload.append({"task_id": task_id, "submitted_answer": scraped_answer})