KWRegan commited on
Commit
c323b30
·
1 Parent(s): 89d5353

Add application file

Browse files
Files changed (1) hide show
  1. 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
+