prozorov commited on
Commit
d3a6234
·
verified ·
1 Parent(s): 6eaace3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -21,7 +21,7 @@ class BasicAgent:
21
  messages = [HumanMessage(content=question)]
22
  messages = self.graph.invoke({"messages": messages})
23
  answer = messages['messages'][-1].content
24
- return answer[14:]
25
 
26
  def run_and_submit_all( profile: gr.OAuthProfile | None):
27
  """
 
21
  messages = [HumanMessage(content=question)]
22
  messages = self.graph.invoke({"messages": messages})
23
  answer = messages['messages'][-1].content
24
+ return answer.split("FINAL ANSWER: ")[1]
25
 
26
  def run_and_submit_all( profile: gr.OAuthProfile | None):
27
  """