Loewolf commited on
Commit
22a827f
·
1 Parent(s): 80dac4b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
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=4,
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=70,
25
- length_penalty=2.0,
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=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
 
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