Spaces:
Sleeping
Sleeping
Commit
·
46f724e
1
Parent(s):
c3e5a9b
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:
|
@@ -38,9 +38,6 @@ def generate(
|
|
38 |
yield output
|
39 |
return output
|
40 |
|
41 |
-
iface = gr.Interface(fn=generate, inputs="text", outputs="text")
|
42 |
-
iface.launch()
|
43 |
-
|
44 |
mychatbot = gr.Chatbot(
|
45 |
avatar_images=["./user.png", "./botm.png"], bubble_full_width=False, show_label=False, show_copy_button=True, likeable=True,)
|
46 |
|
|
|
12 |
return prompt
|
13 |
|
14 |
def generate(
|
15 |
+
prompt, history, temperature=0.2, max_new_tokens=30000, top_p=0.95, repetition_penalty=1.0,
|
16 |
):
|
17 |
temperature = float(temperature)
|
18 |
if temperature < 1e-2:
|
|
|
38 |
yield output
|
39 |
return output
|
40 |
|
|
|
|
|
|
|
41 |
mychatbot = gr.Chatbot(
|
42 |
avatar_images=["./user.png", "./botm.png"], bubble_full_width=False, show_label=False, show_copy_button=True, likeable=True,)
|
43 |
|