oucgc1996 commited on
Commit
bd0e66a
·
verified ·
1 Parent(s): 0b66cfc

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -1,6 +1,7 @@
1
  import random
2
  import torch
3
  import gradio as gr
 
4
  import pandas as pd
5
  from utils import create_vocab, setup_seed
6
  from dataset_mlm import get_paded_token_idx_gen, add_tokens_to_vocab
@@ -121,7 +122,7 @@ iface = gr.Interface(
121
  inputs=[
122
  gr.Slider(minimum=1, maximum=2, step=0.01, label="τ"),
123
  gr.Dropdown(choices=[1,10,100], label="Number of generations"),
124
- gr.RangeSlider(minimum=8, maximum=50, step=1, label="Length Range (Min, Max)")
125
  ],
126
  outputs=["file"]
127
  )
 
1
  import random
2
  import torch
3
  import gradio as gr
4
+ from gradio_rangeslider import RangeSlider
5
  import pandas as pd
6
  from utils import create_vocab, setup_seed
7
  from dataset_mlm import get_paded_token_idx_gen, add_tokens_to_vocab
 
122
  inputs=[
123
  gr.Slider(minimum=1, maximum=2, step=0.01, label="τ"),
124
  gr.Dropdown(choices=[1,10,100], label="Number of generations"),
125
+ range_slider = RangeSlider(minimum=0, maximum=100, value=(0, 100))
126
  ],
127
  outputs=["file"]
128
  )