.

app.py

@@ -49,8 +49,10 @@ def predict_product(num1, num2):
         
         next_token_id = torch.argmax(logits[:, -1, :], dim=-1)
         generated_ids = torch.cat((generated_ids, next_token_id.view(1,-1)), dim=-1)
+        print (next_token_id)
         
         if next_token_id.item() == eos_token_id:
+            print ('berak')
             break
         past_key_values = outputs.past_key_values
         
@@ -113,8 +115,8 @@ demo = gr.Interface(
         gr.Textbox(label='Second Number (up to 12 digits)', value='67890'),
     ],
     outputs=[
-        gr.Textbox(label='Ground Truth Product'),
-        gr.HighlightedText(label='Predicted Product', combine_adjacent=False, show_legend=False, color_map={"-": "green", "+": "red"}),
+        gr.HighlightedText(label='Ground Truth Product', combine_adjacent=False, show_legend=False, color_map={"-": "green", "+": "red"}),
+        gr.HighlightedText(label='GPT2 Predicted Product', combine_adjacent=False, show_legend=False, color_map={"-": "green", "+": "red"}),
         gr.HTML(label='Result Message')
     ],
     title='GPT2 Direct Multiplication Calculator (Without Using Chain-of-Thought)',