Spaces:
Running
on
T4
Running
on
T4
Update app.py
Browse files
app.py
CHANGED
@@ -421,7 +421,11 @@ def generate_beams(n_beams, start_sentence, scores, length_penalty, decoded_sequ
|
|
421 |
|
422 |
@spaces.GPU
|
423 |
def get_beam_search_html(
|
424 |
-
input_text,
|
|
|
|
|
|
|
|
|
425 |
):
|
426 |
inputs = tokenizer([input_text], return_tensors="pt")
|
427 |
|
|
|
421 |
|
422 |
@spaces.GPU
|
423 |
def get_beam_search_html(
|
424 |
+
input_text,
|
425 |
+
number_steps=12,
|
426 |
+
number_beams=1,
|
427 |
+
length_penalty=1.0,
|
428 |
+
num_return_sequences=3
|
429 |
):
|
430 |
inputs = tokenizer([input_text], return_tensors="pt")
|
431 |
|