Update app.py

app.py

@@ -3,6 +3,10 @@ import transformers
 import torch
 import streamlit as st
 
+import re
+
+
+
 model_id = "google/gemma-1.1-2b-it"
 dtype = torch.bfloat16
 
@@ -49,6 +53,9 @@ if prompt := st.chat_input():
     
 
     msg = tokenizer.decode(outputs[0]) #output[0]['generated_text']
+
+    msg = re.sub(r'<.*?>', '', msg)
+    
     
      # Display assistant response in chat message container
     st.chat_message("assistant").write(msg)