Commit
·
e7f4fe4
1
Parent(s):
1c9778f
Update app.py
Browse files
app.py
CHANGED
|
@@ -12,7 +12,7 @@ def format_prompt(message, history):
|
|
| 12 |
return prompt
|
| 13 |
|
| 14 |
def generate(
|
| 15 |
-
prompt, history, temperature=0.2, max_new_tokens=
|
| 16 |
):
|
| 17 |
temperature = float(temperature)
|
| 18 |
if temperature < 1e-2:
|
|
@@ -44,7 +44,7 @@ mychatbot = gr.Chatbot(
|
|
| 44 |
|
| 45 |
demo = gr.ChatInterface(fn=generate,
|
| 46 |
chatbot=mychatbot,
|
| 47 |
-
title="
|
| 48 |
retry_btn=None,
|
| 49 |
undo_btn=None
|
| 50 |
)
|
|
|
|
| 12 |
return prompt
|
| 13 |
|
| 14 |
def generate(
|
| 15 |
+
prompt, history, temperature=0.2, max_new_tokens=512, top_p=0.95, repetition_penalty=1.0,
|
| 16 |
):
|
| 17 |
temperature = float(temperature)
|
| 18 |
if temperature < 1e-2:
|
|
|
|
| 44 |
|
| 45 |
demo = gr.ChatInterface(fn=generate,
|
| 46 |
chatbot=mychatbot,
|
| 47 |
+
title="Kamran's Mixtral 8x7b Chat",
|
| 48 |
retry_btn=None,
|
| 49 |
undo_btn=None
|
| 50 |
)
|