Update app.py

app.py

@@ -39,6 +39,7 @@ def generate(
     repetition_penalty: float = 1.2,
 ) -> Iterator[str]:
     conversation = []
+
     if system_prompt:
         conversation.append({"role": "system", "content": system_prompt})
     for user, assistant in chat_history:
@@ -69,8 +70,15 @@ def generate(
     outputs = []
     for text in streamer:
         outputs.append(text)
-        yield "".join(outputs)
+        # Mask the output here
+        masked_output = mask_with_protecto("".join(outputs))
+        # yield "".join(outputs)
+        yield masked_output
+
 
+def mask_with_protecto(msg_to_mask):
+    return(msg_to_mask)
+    
 
 chat_interface = gr.ChatInterface(
     fn=generate,