Loewolf commited on
Commit
7bef446
1 Parent(s): 0eaf9b2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
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=50,
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=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
 
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