Spaces:
Runtime error
Runtime error
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
|