Add application file
Browse files
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("
|
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")
|