Update app.py

app.py

@@ -16,18 +16,6 @@ model = AutoModelForCausalLM.from_pretrained(model_name_or_path,
 tokenizer = AutoTokenizer.from_pretrained(model_name_or_path, use_fast=True)
 config = MistralConfig()
 
-# Postprocess function
-def postprocess(self, y):
-    if y is None:
-        return []
-    for i, (message, response) in enumerate(y):
-        y[i] = (
-            None if message is None else mdtex2html.convert(message),
-            None if response is None else mdtex2html.convert(response),
-        )
-    return y
-
-gr.Chatbot.postprocess = postprocess
 # Text parsing function
 def _parse_text(text):
     lines = text.split("\n")
@@ -70,7 +58,7 @@ def _launch_demo(args, model, tokenizer, config):
         input_ids = tokenizer.encode(_query, return_tensors='pt')
         
         # Generate a response using the model
-        generated_ids = model.generate(input_ids, max_length=100)
+        generated_ids = model.generate(input_ids, max_length=300)
         
         # Decode the generated IDs to text
         full_response = tokenizer.decode(generated_ids[0], skip_special_tokens=True)