Update app.py

app.py

@@ -13,16 +13,23 @@ def bold_predicted_letters(input_string: str, output_string: str) -> str:
  i = j = 0
  input_string = input_string.upper()
  output_string = output_string.upper()
+ 
  while i < len(input_string):
  if input_string[i:i+6] == "<MASK>":
- result.append("**" + output_string[j] + "**")
- i += 6
- j += 1
+ start_index = i
+ end_index = i + 6
+ while end_index < len(input_string) and input_string[end_index:end_index+6] == "<MASK>":
+ end_index += 6
+ 
+ result.append("**" + output_string[j:j+(end_index-start_index)//6] + "**")
+ i = end_index
+ j += (end_index-start_index)//6
  else:
  result.append(input_string[i])
  i += 1
  if input_string[i-1] != "<":
  j += 1
+ 
  return "".join(result)
 
 class model: