Infinitode Pty Ltd commited on
Commit
c9fc99e
·
verified ·
1 Parent(s): 782b343

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
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
- generated_name = generated_name[0].upper() + generated_name[1:]
 
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()