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 |
|