diff --git a/inference/code-llama/code_completion_example.py b/inference/code-llama/code_completion_example.py
index 106c54ad655077da79df9295ed261c75161856d2..348dadee228992f2f4c34b4ad157504ea423bbe2 100644
--- a/inference/code-llama/code_completion_example.py
+++ b/inference/code-llama/code_completion_example.py
@@ -42,9 +42,7 @@ def main(
             prompt_file
         ), f"Provided Prompt file does not exist {prompt_file}"
         with open(prompt_file, "r") as f:
-            user_prompt = "\n".join(f.readlines())
-    elif not sys.stdin.isatty():
-        user_prompt = "\n".join(sys.stdin.readlines())
+            user_prompt = f.read()
     else:
         print("No user prompt provided. Exiting.")
         sys.exit(1)