Update app.py

app.py

@@ -77,7 +77,7 @@ class ChatBot:
         with torch.no_grad():
             outputs = self.model.generate(
                 **inputs,
-                max_new_tokens=1024,
+                max_new_tokens=2048,
                 temperature=0.7,
                 do_sample=True,
                 pad_token_id=self.tokenizer.eos_token_id,
@@ -99,7 +99,7 @@ bot = ChatBot()
 # Create interface
 with gr.Blocks() as demo:
     gr.HTML('<h1 style="margin:0;">A Bilingual Irish-English LLM — Developed by Abair.ie</h1>')
-    chatbot = gr.Chatbot(height=500)
+    chatbot = gr.Chatbot(height=400)
     msg = gr.Textbox(placeholder="Type your message...", show_label=False)
 
     msg.submit(bot.chat, [msg, chatbot], [chatbot]).then(lambda: "", outputs=msg)