Update app.py

app.py

@@ -9,7 +9,7 @@ import os
 import torch.nn.functional as F
 
 num_return_sequences = 1
-max_length = 200
+max_length = 100
 
 
 @dataclass
@@ -87,5 +87,11 @@ def chat_fn(message, history):
 
 
 gr.ChatInterface(chat_fn, examples=examples,
+                 chatbot=gr.Chatbot(height=300),
+                 textbox=gr.Textbox(placeholder="Ask me a yes or no question", container=False, scale=7),
+                 theme="soft",
+                 cache_examples=True,
+                 retry_btn=None,
+                 undo_btn="Delete Previous",
+                 clear_btn="Clear",
                  title="GPT2 trained from scratch on Shakespeare dataset").launch()
-