[fix]: lowercase input and end with a period.

app.py

@@ -45,6 +45,7 @@ def attention_heatmap(input_tokens: List[str], output_tokens: List[str], weights
 @torch.inference_mode()
 def run(input: str) -> Tuple[str, plt.Figure]:
     """Run inference on a single sentence. Returns prediction and attention heatmap."""""
+    input = input.lower().strip().rstrip(".") + "."
     input_tensor = torch.tensor(source_spm.encode(input), dtype=torch.int64)
     output, weights = model.decode(input_tensor, max_decode_length=max(len(input_tensor), 80))
     output = target_spm.decode(output.detach().tolist())