Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -75,10 +75,10 @@ def generate_random_name(interpreter, vocab_size, sp, max_length=10, temperature
|
|
75 |
next_index = int(next_index)
|
76 |
next_token = sp.id_to_piece(next_index)
|
77 |
|
78 |
-
generated_name
|
79 |
|
80 |
# Decode the generated subword tokens into a string
|
81 |
-
decoded_name = sp.decode_pieces(
|
82 |
|
83 |
# Stop if end token is predicted (optional)
|
84 |
if next_token == '' or len(decoded_name) > max_length:
|
|
|
75 |
next_index = int(next_index)
|
76 |
next_token = sp.id_to_piece(next_index)
|
77 |
|
78 |
+
generated_name = sp.decode_pieces(sp.encode_as_pieces(generated_name) + [next_token])
|
79 |
|
80 |
# Decode the generated subword tokens into a string
|
81 |
+
decoded_name = sp.decode_pieces(sp.encode_as_pieces(generated_name))
|
82 |
|
83 |
# Stop if end token is predicted (optional)
|
84 |
if next_token == '' or len(decoded_name) > max_length:
|