Commit
·
119be10
1
Parent(s):
a062fa4
Update app.py
Browse files
app.py
CHANGED
@@ -14,7 +14,7 @@ def format_prompt(message, history):
|
|
14 |
return prompt
|
15 |
|
16 |
def generate(
|
17 |
-
prompt, history, temperature=0.
|
18 |
):
|
19 |
temperature = float(temperature)
|
20 |
if temperature < 1e-2:
|
@@ -85,5 +85,5 @@ gr.ChatInterface(
|
|
85 |
fn=generate,
|
86 |
chatbot=gr.Chatbot(show_label=False, show_share_button=False, show_copy_button=True, likeable=True, layout="panel"),
|
87 |
additional_inputs=additional_inputs,
|
88 |
-
title="
|
89 |
).launch(show_api=True)
|
|
|
14 |
return prompt
|
15 |
|
16 |
def generate(
|
17 |
+
prompt, history, temperature=0.1, max_new_tokens=256, top_p=0.95, repetition_penalty=1.0,
|
18 |
):
|
19 |
temperature = float(temperature)
|
20 |
if temperature < 1e-2:
|
|
|
85 |
fn=generate,
|
86 |
chatbot=gr.Chatbot(show_label=False, show_share_button=False, show_copy_button=True, likeable=True, layout="panel"),
|
87 |
additional_inputs=additional_inputs,
|
88 |
+
title="DorjGPT interface"
|
89 |
).launch(show_api=True)
|