removed the headers from the text output generated by model

app.py

@@ -46,7 +46,8 @@ def xgen(input_text,
     # Decode the output tensors into string
     outputs_decoded = tokenizer.decode(outputs[0], skip_special_tokens=True)
 
-    output_text = outputs_decoded.strip().replace("Assistant:", "")
+    # Remove header from the output
+    output_text = outputs_decoded.replace(header, "").strip()
 
     return output_text