app.py

app.py

@@ -124,12 +124,12 @@ print("Welcome to the LifGenerator CPU script!")
 print("This script generates lif files using a Hugging Face model and greedy decoding.")
 print(f"Input file path: {args.input_file}")
 print(f"Output file path: {args.output_file}")
-INPUT_FILE = args.input_file
+INPUT_FILE = args.input_file if args.input_file else "Kangaroos.txt"
 INPUT_FILE_STEM = INPUT_FILE.split('.')[0]
 OUTPUT_FILE = args.output_file if args.output_file else (INPUT_FILE_STEM + ".lif")
 PROMPT_ID = args.prompt_id if args.prompt_id else INPUT_FILE
 PROMPT_TOPIC = args.prompt_topic if args.prompt_topic else INPUT_FILE
-MULTI_PV = args.multi_pv if args.multi_pv else 100
+MULTI_PV = args.multi_pv if args.multi_pv else 1000
 NUM_WORDS = args.num_words if args.num_words else 10000
 NUM_TOKENS = args.num_tokens if args.num_tokens else 10000
 BEAM_WIDTH = args.beam_width if args.beam_width else 1