app.py

app.py

@@ -122,11 +122,13 @@ parser.add_argument(
 args = parser.parse_args()
 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}")
+#print(f"Input file path: {args.input_file}")
+#print(f"Output file path: {args.output_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")
+print(f"Input file path: {INPUT_FILE}")
+print(f"Output file path: {OUTPUT_FILE}")
 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 1000
@@ -213,8 +215,8 @@ def main(INPUT_FILE, OUTPUT_FILE, PROMPT_ID, PROMPT_TOPIC, MULTI_PV, NUM_WORDS,
     # %% Constants and Configuration
     # MODEL_NAME = "deepseek-ai/DeepSeek-R1-Distill-Qwen-1.5B"
     # MODEL_NAME = "google/gemma-3-4b-it"
-    # MODEL_NAME = "Qwen/Qwen3-1.7B"
-    MODEL_NAME = "Qwen/Qwen3-0.6B"
+    MODEL_NAME = "Qwen/Qwen3-1.7B"
+    # MODEL_NAME = "Qwen/Qwen3-0.6B"
     # MODEL_NAME = "microsoft/Phi-4-mini-instruct"
     #MODEL_NAME = "meta-llama/Llama-2-7b-hf"
     #MODEL_NAME = input(f"Enter hugging face model name or press enter to default to [{MODEL_NAME}]: ") or MODEL_NAME
@@ -281,8 +283,8 @@ def main(INPUT_FILE, OUTPUT_FILE, PROMPT_ID, PROMPT_TOPIC, MULTI_PV, NUM_WORDS,
     # LegalNumberOfMove_str = input("  Enter Max Number of moves [Default: 10]: ") or "10"
     LegalNumberOfMove = NUM_WORDS  # int(LegalNumberOfMove_str)
     # EngineID = f"DeepSeek R1 1.5B Qwen-Distil Greedy ({DEVICE.upper()})"  # Updated EngineID
-    # EngineID = f"Qwen/Qwen3-1.7B"
-    EngineID = f"Qwen/Qwen3-0.6B"
+    EngineID = f"Qwen/Qwen3-1.7B"
+    # EngineID = f"Qwen/Qwen3-0.6B"
     # EngineID = f"Gemma-3-4b-it ({DEVICE.upper()})" # Indicate CPU in EngineID
     Depth = 1
     print("  User inputs captured.\n")
@@ -588,6 +590,7 @@ def main(INPUT_FILE, OUTPUT_FILE, PROMPT_ID, PROMPT_TOPIC, MULTI_PV, NUM_WORDS,
 def premain():
    main("Kangaroos.txt", "KangaroosByQ06.lif", "KangarooEssay", "The Mating Habits of Kangaroos", 100, 1000, 10000, 1, 0, 1, "Qwen")
 
-demo = gr.Interface(fn=premain, inputs="text", outputs="text")
-demo.launch()
+#demo = gr.Interface(fn=premain, inputs="text", outputs="text")
+demo = gr.Interface(fn=premain)
+demo.launch(share=True)