Update app.py

app.py

@@ -58,6 +58,14 @@ client_s = Client(
     API_URL_S, headers={"Authorization": f"Bearer {HF_TOKEN}"},
 )
 
+def wrap_html_code(text):
+    pattern = r"<.*?>"
+    matches = re.findall(pattern, text)
+    if len(matches) > 0:
+        return f"```{text}```"
+    else:
+        return text
+        
 def generate(
     prompt,
     temperature=0.9,
@@ -122,7 +130,7 @@ def generate(
             #t += 1
         previous_token = response.token.text
     print("Output = "+str(output))
-    return output.strip()
+    return wrap_html_code(output.strip())
 
 
 # chatbot mode