Update app.py
Browse files
app.py
CHANGED
@@ -73,7 +73,7 @@ examples = [
|
|
73 |
starling_bot = StarlingBot()
|
74 |
|
75 |
def gradio_starling(user_message, assistant_message, system_prompt, do_sample, temperature, max_new_tokens, top_p, repetition_penalty):
|
76 |
-
response = starling_bot.predict(user_message, assistant_message,
|
77 |
return response
|
78 |
|
79 |
with gr.Blocks(theme="ParityError/Anime") as demo:
|
@@ -98,7 +98,7 @@ with gr.Blocks(theme="ParityError/Anime") as demo:
|
|
98 |
|
99 |
submit_button.click(
|
100 |
gradio_starling,
|
101 |
-
inputs=[user_message, assistant_message,
|
102 |
outputs=output_text
|
103 |
)
|
104 |
|
|
|
73 |
starling_bot = StarlingBot()
|
74 |
|
75 |
def gradio_starling(user_message, assistant_message, system_prompt, do_sample, temperature, max_new_tokens, top_p, repetition_penalty):
|
76 |
+
response = starling_bot.predict(user_message, assistant_message, system_prompt, do_sample, temperature, max_new_tokens, top_p, repetition_penalty)
|
77 |
return response
|
78 |
|
79 |
with gr.Blocks(theme="ParityError/Anime") as demo:
|
|
|
98 |
|
99 |
submit_button.click(
|
100 |
gradio_starling,
|
101 |
+
inputs=[user_message, assistant_message, system_prompt, do_sample, temperature, max_new_tokens, top_p, repetition_penalty],
|
102 |
outputs=output_text
|
103 |
)
|
104 |
|