Update app.py
Browse files
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 |
-
|
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 |
)
|