Update app.py

app.py

@@ -74,7 +74,10 @@ def bot(input_message: str, temperature=0.1, top_p=0.9, top_k=0, repetition_pena
     for new_text in streamer:
         partial_text += new_text
 
-    return partial_text
+    # Split the text by "|", and get the last element in the list which should be the final query
+    final_query = partial_text.split("|")[-1].strip()
+    return final_query
+
 
 gradio_interface = gr.Interface(
   fn=bot,