Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -21,8 +21,8 @@ def generate_text(prompt):
|
|
21 |
early_stopping=True,
|
22 |
temperature=0.6,
|
23 |
top_p=0.95,
|
24 |
-
top_k=
|
25 |
-
|
26 |
do_sample=True,
|
27 |
eos_token_id=tokenizer.eos_token_id,
|
28 |
pad_token_id=tokenizer.eos_token_id
|
@@ -60,7 +60,7 @@ h1 {
|
|
60 |
|
61 |
iface = gr.Interface(
|
62 |
fn=generate_text,
|
63 |
-
inputs=gr.Textbox(lines=
|
64 |
outputs=gr.Textbox(label="Löwolf Chat Responses", placeholder="Responses will appear here...", interactive=False, lines=10),
|
65 |
|
66 |
css=css
|
|
|
21 |
early_stopping=True,
|
22 |
temperature=0.6,
|
23 |
top_p=0.95,
|
24 |
+
top_k=10,
|
25 |
+
repetition_penalty: 1.2,
|
26 |
do_sample=True,
|
27 |
eos_token_id=tokenizer.eos_token_id,
|
28 |
pad_token_id=tokenizer.eos_token_id
|
|
|
60 |
|
61 |
iface = gr.Interface(
|
62 |
fn=generate_text,
|
63 |
+
inputs=gr.Textbox(lines=2, placeholder="Type a message...", label="Your Message"),
|
64 |
outputs=gr.Textbox(label="Löwolf Chat Responses", placeholder="Responses will appear here...", interactive=False, lines=10),
|
65 |
|
66 |
css=css
|