CreitinGameplays commited on
Commit
e178ac6
1 Parent(s): a4e9281

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -5
app.py CHANGED
@@ -22,11 +22,11 @@ def generate_text(user_prompt):
22
  input_ids=prompt_encoded,
23
  max_length=256,
24
  num_beams=1,
25
- num_return_sequences=1, # Generate only 1 sequence
26
- do_sample=True, # Enable sampling for creativity
27
- top_k=50, # Sample from the top 50 most likely tokens at each step
28
- top_p=0.15, # Filter out highly probable unlikely continuations
29
- temperature=0.1, # Control the randomness of the generated text (1.0 for default)
30
  repetition_penalty=1.165
31
  )
32
 
 
22
  input_ids=prompt_encoded,
23
  max_length=256,
24
  num_beams=1,
25
+ num_return_sequences=1,
26
+ do_sample=True,
27
+ top_k=0,
28
+ top_p=0.95,
29
+ temperature=0.2,
30
  repetition_penalty=1.165
31
  )
32