Added code markdown output

app.py

@@ -1,6 +1,6 @@
 import gradio as gr
 from transformers import TFT5ForConditionalGeneration, RobertaTokenizer
-import regex as re
+
 # load saved finetuned model
 model = TFT5ForConditionalGeneration.from_pretrained('Yati05/TF-CodeT5-base')
 # load saved tokenizer
@@ -19,7 +19,7 @@ def chat(chat_history, user_input):
     # decode generated tokens
     decoded_code = tokenizer.decode(generated_code.numpy()[0], skip_special_tokens=True)
     
-    return chat_history + [(user_input, re.sub(r"( {4})", " ", decoded_code))]
+    return chat_history + [(user_input, "<pre><code>"+decoded_code+"</code></pre>")]
 #   response = ""
 #   for letter in decoded_code: #[decoded_code[i:i+1] for i in range(0, len(decoded_code), 1)]:
 #       response += letter + ""