teragron commited on
Commit
ed5c215
·
1 Parent(s): e295ff0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -3,7 +3,7 @@ import gradio as gr
3
 
4
  #model_file = "best_model_4800it_40M.bin"
5
 
6
-
7
 
8
  try:
9
  subprocess.run(["make", "runfast"], check=True, shell=True)
@@ -31,7 +31,7 @@ with gr.Blocks() as demo:
31
  temperature_slider = gr.Slider(minimum=0.1, maximum=2.0, value=0.8, label="Temperature")
32
  topt_slider = gr.Slider(minimum=0.1, maximum=1.0, value=0.9, label="Topt")
33
  maxtoken_slider = gr.Slider(minimum=64, maximum=1024, value=256, label="Max Tokens")
34
- model_pick = gr.Dropdown(["best_model_4800it_40M.bin", "model_2200.bin", "model_4200.bin", "model_2200_tr_corpus.bin", "model_8k.bin", "model_1600.bin"], label="Model", info="Model")
35
  out = gr.Textbox()
36
 
37
  btn = gr.Button("Run")
 
3
 
4
  #model_file = "best_model_4800it_40M.bin"
5
 
6
+ model_list = ["best_model_4800it_40M.bin", "model_2200.bin", "model_4200.bin", "model_2200_tr_corpus.bin", "model_8k.bin", "model_1600.bin", "model_3600.bin"]
7
 
8
  try:
9
  subprocess.run(["make", "runfast"], check=True, shell=True)
 
31
  temperature_slider = gr.Slider(minimum=0.1, maximum=2.0, value=0.8, label="Temperature")
32
  topt_slider = gr.Slider(minimum=0.1, maximum=1.0, value=0.9, label="Topt")
33
  maxtoken_slider = gr.Slider(minimum=64, maximum=1024, value=256, label="Max Tokens")
34
+ model_pick = gr.Dropdown(model_list, label="Model", info="Model")
35
  out = gr.Textbox()
36
 
37
  btn = gr.Button("Run")