Update app.py

app.py

@@ -88,9 +88,9 @@ def generate_random_name(interpreter, vocab_size, sp, max_length=10, temperature
     decoded_name = decoded_name.replace("</s>", "")
     decoded_name = decoded_name.replace("<unk>", "")
     decoded_name = decoded_name.replace("<s>", "")
-    generated_name = decoded_name.rsplit(' ', 1)[0]
-    if generated_name:
-        generated_name = generated_name[0].upper() + generated_name[1:]
+    generated_name = decoded_name.strip()
+
+    generated_name = generated_name.capitalize()
 
     # Split the name and check the last part
     parts = generated_name.split()