added formatted output

app.py

@@ -154,6 +154,8 @@ async def infer(request: InferenceRequest):
     
     pred = preds[0].cpu().numpy()
     pred_nl = tokenizer.decode(pred[2:], skip_special_tokens=True, clean_up_tokenization_spaces=False)
+    # Replace <add> markers with newlines and strip whitespace
+    pred_nl = "\n".join([seg.strip() for seg in pred_nl.split("<add>") if seg.strip()])
     last_infer_result = {"generated_code": pred_nl}
     return last_infer_result