fix the extra line added and remove take prompt from cli

jondurbin · Aug 27, 2023 · 75991d8 · 75991d8
1 parent d28fc98
commit 75991d8
Showing 1 changed file with 1 addition and 3 deletions.
diff --git 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)