Add application file
Browse files- lifgen-hook.py +11 -0
lifgen-hook.py
CHANGED
@@ -10,6 +10,14 @@
|
|
10 |
#
|
11 |
# Example:
|
12 |
# python lifgen-hook.py -i YaoJokic.txt -o YaoJokicm50.lif -pid YaoTestByDeepSeek -pt "Compare Yao Ming and Nikola Jokic in NBA basketball" -mpv 50
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
13 |
import math
|
14 |
|
15 |
import torch
|
@@ -576,3 +584,6 @@ def main(INPUT_FILE, OUTPUT_FILE, PROMPT_ID, PROMPT_TOPIC, MULTI_PV, NUM_WORDS,
|
|
576 |
main(INPUT_FILE, OUTPUT_FILE, PROMPT_ID, PROMPT_TOPIC, MULTI_PV, NUM_WORDS, NUM_TOKENS, BEAM_WIDTH, ALPHA_MODE,
|
577 |
MODEL_TAG)
|
578 |
|
|
|
|
|
|
|
|
10 |
#
|
11 |
# Example:
|
12 |
# python lifgen-hook.py -i YaoJokic.txt -o YaoJokicm50.lif -pid YaoTestByDeepSeek -pt "Compare Yao Ming and Nikola Jokic in NBA basketball" -mpv 50
|
13 |
+
|
14 |
+
import gradio as gr
|
15 |
+
|
16 |
+
|
17 |
+
|
18 |
+
|
19 |
+
|
20 |
+
|
21 |
import math
|
22 |
|
23 |
import torch
|
|
|
584 |
main(INPUT_FILE, OUTPUT_FILE, PROMPT_ID, PROMPT_TOPIC, MULTI_PV, NUM_WORDS, NUM_TOKENS, BEAM_WIDTH, ALPHA_MODE,
|
585 |
MODEL_TAG)
|
586 |
|
587 |
+
demo = gr.Interface(fn=main, inputs="text", outputs="text")
|
588 |
+
demo.launch()
|
589 |
+
|