Update app.py

app.py

@@ -112,7 +112,12 @@ TcDebug (0) > NONE??? LtacNotationCall (no location),/mnt/data1/2024/01/05/UniMa
 """
 # source : unimath/unimath/batch2/data_42/BATCH142042
 
-unimath_examples = [unimath1, unimath2, unimath3, unimath4]
+examples = [
+    (unimath1, additional_info_prompt),
+    (unimath2, additional_info_prompt),
+    (unimath3, additional_info_prompt),
+    (unimath4, additional_info_prompt),
+]
 
 model_name = "deepseek-ai/deepseek-math-7b-instruct"
 tokenizer = AutoTokenizer.from_pretrained(model_name)
@@ -159,15 +164,12 @@ def main():
             additional_info = additional_info_prompt
             return question, additional_info
 
-        with gr.Accordion("🤖 UniMath Examples", open=False):
-            gr.Markdown("Click on a 🤖 UniMath Example to load it into the solver.")
-            for i, _ in enumerate([unimath1, unimath2, unimath3, unimath4]):
-                example_button = gr.Button(f"Example {i+1}")
-                example_button.click(fn=set_example, inputs=[i], outputs=[question_field, additional_info_field])
+        with gr.Row():
+            with gr.Accordion("🤖 UniMath Examples", open=False):
+                for i, example in enumerate(examples, start=1):
+                    gr.Markdown(f"**Example {i}:**\n```{example[0]}```")
 
-        submit_button.click(fn=solve_math_problem, inputs=[question_field, additional_info_field, max_tokens], outputs=[full_answer_output, final_answer_output])
-
-    demo.launch()
+    demo.launch(examples=examples)
 
 if __name__ == "__main__":
     main()