Update app.py

app.py

@@ -14,8 +14,11 @@ def format_prompt(message, history):
   return prompt
 
 def generate(
-    prompt, history, temperature=1.2, max_new_tokens=1024, top_p=0.95, repetition_penalty=1.0,
+    prompt, history, system_prompt="", max_new_tokens=1024, temperature=1.2, top_p=0.95, repetition_penalty=1.0,
 ):
+    print(history)
+    print(system_prompt)
+    print(max_new_tokens)
     temperature = float(temperature)
     if temperature < 1e-2:
         temperature = 1e-2
@@ -48,7 +51,11 @@ demo = gr.ChatInterface(fn=generate,
                         chatbot=mychatbot,
                         title="fast ai aaaaa!!",
                         retry_btn="Regenerate",
-                        undo_btn="Undo"
+                        undo_btn="Undo",
+                        additional_inputs=[
+                            gr.Textbox(label="System prompt", lines=3)
+                            gr.Slider(label="Max new tokens", maximum=2048, value=512)
+                        ]
                        )
 
 demo.queue().launch(show_api=True)