Spaces:
Runtime error
Runtime error
rm contrastive search since it needs a lot of memory
Browse files
app.py
CHANGED
@@ -54,7 +54,7 @@ with gr.Blocks(css=css) as demo:
|
|
54 |
user_input = gr.inputs.Textbox(placeholder="",
|
55 |
label="Ask me something",
|
56 |
default="How old is the universe?")
|
57 |
-
decoding_method = gr.inputs.Dropdown(["Beam Search", "Sampling"
|
58 |
default="Sampling", label="Decoding Method")
|
59 |
num_beams = gr.inputs.Slider(label="Number of beams for beam search",
|
60 |
default=1, minimum=1, maximum=10, step=1)
|
|
|
54 |
user_input = gr.inputs.Textbox(placeholder="",
|
55 |
label="Ask me something",
|
56 |
default="How old is the universe?")
|
57 |
+
decoding_method = gr.inputs.Dropdown(["Beam Search", "Sampling"],
|
58 |
default="Sampling", label="Decoding Method")
|
59 |
num_beams = gr.inputs.Slider(label="Number of beams for beam search",
|
60 |
default=1, minimum=1, maximum=10, step=1)
|