Update app.py

app.py

@@ -5,18 +5,26 @@ prompt=''' Review the Following text as a human, who is tasked to Extract `Code
 
 "/{}/"
 '''
-
+replacements = {
+    '{': 'OpenCurlyBrace',
+    '}': 'CloseCurlyBrace',
+    '<': 'LessThanSign'
+}
+​
 model = AutoModelForSeq2SeqLM.from_pretrained("google/flan-t5-large")
 tokenizer = AutoTokenizer.from_pretrained("google/flan-t5-large")
 
 def chatbot_model(m,history):
-   
+     for char, replacement in replacements.items():
+        m = m.replace(char, replacement)
     # Encode the prompt and generate response
     input_ids = tokenizer.encode(prompt.format(m) + tokenizer.eos_token, return_tensors='pt')
     outputs = model.generate(input_ids, max_length=1000, pad_token_id=tokenizer.eos_token_id)
-
+    
     # Decode the output
     decoded_output = tokenizer.decode(outputs[0], skip_special_tokens=True)
+    for char, replacement in replacements.items():
+        decoded_outputs = decoded_outputs.replace(replacement, char)    
     for i in range(len(decoded_output)):
         time.sleep(0.3)
         yield decoded_output[:i+1]