Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -15,14 +15,14 @@ def generate_text(prompt):
|
|
15 |
input_ids,
|
16 |
attention_mask=attention_mask,
|
17 |
max_length=max_length,
|
18 |
-
min_length=
|
19 |
num_beams=5,
|
20 |
no_repeat_ngram_size=2,
|
21 |
early_stopping=True,
|
22 |
temperature=0.7,
|
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
|
|
|
15 |
input_ids,
|
16 |
attention_mask=attention_mask,
|
17 |
max_length=max_length,
|
18 |
+
min_length=1,
|
19 |
num_beams=5,
|
20 |
no_repeat_ngram_size=2,
|
21 |
early_stopping=True,
|
22 |
temperature=0.7,
|
23 |
top_p=0.95,
|
24 |
+
top_k=50,
|
25 |
+
|
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=1, 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
|