Update app.py

Added a sanitization for the <think> tags

app.py

@@ -62,6 +62,12 @@ def chat_fn(history, enable_thinking, temperature, top_p, top_k, repetition_pena
     )
     output_ids = streamer[0][len(inputs.input_ids[0]):]
     response = tokenizer.decode(output_ids, skip_special_tokens=True)
+    if isinstance(response, str): 
+        response = response.replace('<think>',"# &lt;think&gt;").replace('</think>',"&lt;/think&gt;")
+    elif isinstance(response,list):
+        response = [paper.replace('<think>',"# &lt;think&gt;").replace('</think>',"&lt;/think&gt;") for paper in response]
+    else:
+        raise ValueError("Tokenizer response seems malformed; Not a string, nor a list?!?!")
 
     # streaming char-by-char
     history.append({"role": "assistant", "content": ""})