Update app.py

app.py

@@ -229,9 +229,8 @@ class CharTokenizer:
         return ids + [self.pad_token_id] * (max_length - len(ids))
 
     def decode(self, ids):
-        return "".join(
-            id2char[i] for i in ids if i != self.pad_token_id
-        )
+        return "".join(id2char.get(i, "") for i in ids if i != self.pad_token_id)
+
 
 tokenizer = CharTokenizer()