Spaces:
Sleeping
Sleeping
Infinitode Pty Ltd
commited on
Update app.py
Browse files
app.py
CHANGED
|
@@ -88,7 +88,8 @@ def generate_random_name(interpreter, vocab_size, sp, max_length=10, temperature
|
|
| 88 |
decoded_name = decoded_name.replace("<unk>", "")
|
| 89 |
decoded_name = decoded_name.replace("<s>", "")
|
| 90 |
generated_name = decoded_name.rsplit(' ', 1)[0]
|
| 91 |
-
|
|
|
|
| 92 |
|
| 93 |
# Split the name and check the last part
|
| 94 |
parts = generated_name.split()
|
|
|
|
| 88 |
decoded_name = decoded_name.replace("<unk>", "")
|
| 89 |
decoded_name = decoded_name.replace("<s>", "")
|
| 90 |
generated_name = decoded_name.rsplit(' ', 1)[0]
|
| 91 |
+
if generated_name:
|
| 92 |
+
generated_name = generated_name[0].upper() + generated_name[1:]
|
| 93 |
|
| 94 |
# Split the name and check the last part
|
| 95 |
parts = generated_name.split()
|