diff --git a/torchchat/edge/android/torchchat/app/src/main/java/org/pytorch/torchchat/MainActivity.java b/torchchat/edge/android/torchchat/app/src/main/java/org/pytorch/torchchat/MainActivity.java
index 4be7303c3..2178d6c05 100644
--- a/torchchat/edge/android/torchchat/app/src/main/java/org/pytorch/torchchat/MainActivity.java
+++ b/torchchat/edge/android/torchchat/app/src/main/java/org/pytorch/torchchat/MainActivity.java
@@ -204,7 +204,7 @@ private void onModelRunStopped() {
         mSendButton.setText("Generate");
         mSendButton.setOnClickListener(
                 view -> {
-                    String prompt = mEditTextMessage.getText().toString();
+                    String prompt = mEditTextMessage.getText().toString() + "<|eot_id|><|start_header_id|>assistant<|end_header_id|>\n\n"
                     mMessageAdapter.add(new Message(prompt, true));
                     mMessageAdapter.notifyDataSetChanged();
                     mEditTextMessage.setText("");