.

app.py

@@ -31,7 +31,7 @@ def predict_product(num1, num2):
     raw_output = tokenizer.decode(output, skip_special_tokens=True)
     prediction = postprocess(raw_output)
 
-    # Evalaute the correctness of the result
+    # Evaluate the correctness of the result
     try:
         num1_int = int(num1)
         num2_int = int(num2)
@@ -52,7 +52,7 @@ def predict_product(num1, num2):
         result_message = "Invalid input. Could not evaluate correctness."
     result_html = f"<div style='color: {result_color};'>{result_message}</div>"
 
-    return input_text, raw_output, prediction, result_html
+    return prediction, result_html
 
 demo = gr.Interface(
     fn=predict_product,
@@ -61,8 +61,6 @@ demo = gr.Interface(
         gr.Textbox(label='Second Number (up to 12 digits)', value='67890'),
     ],
     outputs=[
-        gr.Textbox(label='Raw Input to GPT2 (reversed digits and added spaces)'),
-        gr.Textbox(label='Raw Output from GPT2 (reversed digits and with spaces)'),
         gr.Textbox(label='Predicted Product'),
         gr.HTML(label='Result Message')
     ],