Update app.py

app.py

@@ -37,13 +37,11 @@ def predict(input_text):
     # assemble the array back to a string
     pcnt_file_cr=' '.join(pcnt_file_array_hash).replace('#','\n')
 
-    
-    p = re.compile(r'(?<=[\.\?!]\s)(\w+)')
-    pcnt_file_cr_cap = p.sub(cap, pcnt_file_cr)
+    regex1 = r"\bi\b"
+    regex2 = r"(?<=[.?!]|\n)\s*\w"
+    pcnt_file_cr_cap = re.sub(regex2, lambda x: x.group().upper(), re.sub(regex1, "I", pcnt_file_cr))
 
-    pcnt_file_cr_capi = pcnt_file_cr_cap.replace(' i ',' I ')
-    
-    return pcnt_file_cr_capi
+    return pcnt_file_cr_cap
  
 if __name__ == "__main__":