.

app.py

@@ -26,7 +26,6 @@ def predict_answer(question):
 
     return prediction
 
-color_map = {"correct": "green", "wrong": "red"}
 
 demo = gr.Interface(
     fn=predict_answer,
@@ -34,7 +33,7 @@ demo = gr.Interface(
         gr.Textbox(label='Question', value='A set of 7 spoons costs $21. If each spoon would be sold separately, how much would 5 spoons cost?'),
     ],
     outputs=[
-        gr.HighlightedText(label='Implicit CoT Prediction', combine_adjacent=False, show_legend=False, color_map=color_map, show_inline_category=False),
+        gr.Textbox(label='Implicit CoT Prediction'),
     ],
     title='Solving Grade School Math Problems with Implicit CoT',
     description='This demo showcases Mistral-7B\'s ability to solve grade school math problems without producing intermediate steps, using our stepwise internalization method.',