Update app.py

app.py

@@ -239,10 +239,13 @@ def respond(
     history: list[tuple[str, str]],
     available_pic_str,
     system_message,
+    su_msg,
     max_tokens,
     temperature,
     top_p
 ):
+    SUS_PROMPT = su_msg
+    
     for val in history:
         if "[END OF CHAT]" in val:
             return "[END OF CHAT]"
@@ -287,6 +290,7 @@ demo = gr.ChatInterface(
     additional_inputs=[
         gr.Textbox(value=AVAILABLE_PICTURES, label="Available pics (follow format)"),
         gr.Textbox(value=SYSTEM_PROMPT, label="System message"),
+        gr.Textbox(value=SUS_PROMPT, label="SU Message"),
         gr.Slider(minimum=1, maximum=2048, value=512, step=1, label="Max new tokens"),
         gr.Slider(minimum=0.1, maximum=4.0, value=0.7, step=0.1, label="Temperature"),
         gr.Slider(