Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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(
|
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")
|