KWRegan commited on
Commit
b1af265
·
1 Parent(s): c90eb34

Add application file

Browse files
Files changed (1) hide show
  1. app.py +5 -3
app.py CHANGED
@@ -216,7 +216,8 @@ def main(INPUT_FILE, OUTPUT_FILE, PROMPT_ID, PROMPT_TOPIC, MULTI_PV, NUM_WORDS,
216
  # %% Constants and Configuration
217
  # MODEL_NAME = "deepseek-ai/DeepSeek-R1-Distill-Qwen-1.5B"
218
  # MODEL_NAME = "google/gemma-3-4b-it"
219
- MODEL_NAME = "Qwen/Qwen3-1.7B"
 
220
  # MODEL_NAME = "Qwen/Qwen3-0.6B"
221
  # MODEL_NAME = "microsoft/Phi-4-mini-instruct"
222
  #MODEL_NAME = "meta-llama/Llama-2-7b-hf"
@@ -285,7 +286,8 @@ def main(INPUT_FILE, OUTPUT_FILE, PROMPT_ID, PROMPT_TOPIC, MULTI_PV, NUM_WORDS,
285
  # LegalNumberOfMove_str = input(" Enter Max Number of moves [Default: 10]: ") or "10"
286
  LegalNumberOfMove = NUM_WORDS # int(LegalNumberOfMove_str)
287
  # EngineID = f"DeepSeek R1 1.5B Qwen-Distil Greedy ({DEVICE.upper()})" # Updated EngineID
288
- EngineID = f"Qwen/Qwen3-1.7B"
 
289
  # EngineID = f"Qwen/Qwen3-0.6B"
290
  # EngineID = f"Gemma-3-4b-it ({DEVICE.upper()})" # Indicate CPU in EngineID
291
  Depth = 1
@@ -595,7 +597,7 @@ def main(INPUT_FILE, OUTPUT_FILE, PROMPT_ID, PROMPT_TOPIC, MULTI_PV, NUM_WORDS,
595
 
596
  def premain(ip):
597
  #txtoutput = main("Kanga.txt", "KangaroosByQ06.lif", "KangarooEssay", "The Mating Habits of Kangaroos", 100, 1000, 10000, 1, 0, 1, "Qwen")
598
- txtoutput = main("Kanga20.txt", "Kanga20ByQ06.lif", "KangarooEssay", "The Mating Habits of Kangaroos", 100, 1000, 10000, 1, 0, 1, "Qwen")
599
  return txtoutput
600
 
601
  demo = gr.Interface(fn=premain, inputs="text", outputs="text")
 
216
  # %% Constants and Configuration
217
  # MODEL_NAME = "deepseek-ai/DeepSeek-R1-Distill-Qwen-1.5B"
218
  # MODEL_NAME = "google/gemma-3-4b-it"
219
+ # MODEL_NAME = "Qwen/Qwen3-1.7B"
220
+ MODEL_NAME = "Qwen/Qwen3-4B"
221
  # MODEL_NAME = "Qwen/Qwen3-0.6B"
222
  # MODEL_NAME = "microsoft/Phi-4-mini-instruct"
223
  #MODEL_NAME = "meta-llama/Llama-2-7b-hf"
 
286
  # LegalNumberOfMove_str = input(" Enter Max Number of moves [Default: 10]: ") or "10"
287
  LegalNumberOfMove = NUM_WORDS # int(LegalNumberOfMove_str)
288
  # EngineID = f"DeepSeek R1 1.5B Qwen-Distil Greedy ({DEVICE.upper()})" # Updated EngineID
289
+ # EngineID = f"Qwen/Qwen3-1.7B"
290
+ EngineID = f"Qwen/Qwen3-4B"
291
  # EngineID = f"Qwen/Qwen3-0.6B"
292
  # EngineID = f"Gemma-3-4b-it ({DEVICE.upper()})" # Indicate CPU in EngineID
293
  Depth = 1
 
597
 
598
  def premain(ip):
599
  #txtoutput = main("Kanga.txt", "KangaroosByQ06.lif", "KangarooEssay", "The Mating Habits of Kangaroos", 100, 1000, 10000, 1, 0, 1, "Qwen")
600
+ txtoutput = main("Kangaroos.txt", "KangaroosByQ4.lif", "KangarooEssay", "The Mating Habits of Kangaroos", 1000, 1000, 10000, 1, 0, 1, "Qwen4B")
601
  return txtoutput
602
 
603
  demo = gr.Interface(fn=premain, inputs="text", outputs="text")