Debug output

app.py

@@ -41,8 +41,10 @@ def predict(type, normalized_asm):
         no_mask_idx=torch.LongTensor([tokenizer_output['no_mask_idx']]).cuda(), 
         pad_token_id=tokenizer.pad_token_id, eos_token_id=tokenizer.eos_token_id
     )
+    print("Output 1:", output)
 
     output = tokenizer.decode(output[input_ids.size(1): ], skip_special_tokens=True, clean_up_tokenization_spaces=True)
+    print("Output 2:", output)
 
     return output