Update app.py

app.py

@@ -75,8 +75,8 @@ examples = [
 # Initialize StarlingBot
 starling_bot = StarlingBot()
 
-def gradio_starling(user_message, assistant_message, system_prompt, do_sample, temperature, max_new_tokens, top_p, repetition_penalty):
-    response = starling_bot.predict(user_message, assistant_message, system_prompt, do_sample, temperature, max_new_tokens, top_p, repetition_penalty)
+def gradio_starling(user_message, assistant_message, system_prompt, mode, do_sample, temperature, max_new_tokens, top_p, repetition_penalty):
+    response = starling_bot.predict(user_message, assistant_message, system_prompt, mode, do_sample, temperature, max_new_tokens, top_p, repetition_penalty)
     return response
 
 with gr.Blocks(theme="ParityError/Anime") as demo: