app.py

app.py

@@ -210,7 +210,7 @@ else:
     print("HF Token not found. Gated model download might fail.")
 
 
-def main(INPUT_FILE, OUTPUT_FILE, PROMPT_ID, PROMPT_TOPIC, MULTI_PV, NUM_WORDS, NUM_TOKENS, BEAM_WIDTH, ALPHA_MODE,
+def main(INPUT_FILE, OUTPUT_FILE, PROMPT_ID, PROMPT_TOPIC, MULTI_PV, NUM_WORDS, NUM_TOKENS, BEAM_WIDTH, ALPHA_MODE, START_TURN,
          MODEL_TAG):
     # %% Constants and Configuration
     # MODEL_NAME = "deepseek-ai/DeepSeek-R1-Distill-Qwen-1.5B"
@@ -587,10 +587,10 @@ def main(INPUT_FILE, OUTPUT_FILE, PROMPT_ID, PROMPT_TOPIC, MULTI_PV, NUM_WORDS,
 #main(INPUT_FILE, OUTPUT_FILE, PROMPT_ID, PROMPT_TOPIC, MULTI_PV, NUM_WORDS, NUM_TOKENS, BEAM_WIDTH, ALPHA_MODE,
      #START_TURN, MODEL_TAG)
 
-def premain(ip, op):
+def premain(ip):
    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 = gr.Interface(fn=premain)
-demo.launch(share=True)
+demo.launch()