Update app.py

app.py

@@ -68,13 +68,7 @@ def generate(
     t = Thread(target=model.generate, kwargs=generate_kwargs)
     t.start()
 
-#    outputs = []
-#    for text in streamer:
-#        outputs.append(text)
-#    concatenated_outputs = " ".join(outputs)
-    # yield concatenated_outputs
     concatenated_outputs = "".join([r"{}".format(text) for text in streamer])
-#    yield concatenated_outputs
 
     # Mask the output here
     masked_output = mask_with_protecto(concatenated_outputs)
@@ -82,14 +76,12 @@ def generate(
     yield masked_output
 
 
+# Ensuring entity tags are properly rendered
 def format_for_html(text):
     text = text.replace("<", "&lt;")
     text = text.replace(">", "&gt;")
     return text
 
-# Using the function
-api_output = "N7JI9bzqbf M1L4LPGkyj is the CEO of Apple Inc."
-formatted_output = format_for_html(api_output)
     
 def mask_with_protecto(text_for_prompt):
     mask_request_url = "https://trial.protecto.ai/api/vault/mask"