smgriffin commited on
Commit
c1f3433
·
1 Parent(s): ec9a94a

more text generated

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -12,7 +12,7 @@ def generate_lyrics(artist):
12
  input_ids = tokenizer.encode(prompt, return_tensors="pt")
13
  output = model.generate(
14
  input_ids,
15
- max_length=150, # Maximum number of tokens
16
  temperature=0.9, # Adjust randomness
17
  top_k=50, # Top-k sampling
18
  top_p=0.95, # Nucleus sampling
 
12
  input_ids = tokenizer.encode(prompt, return_tensors="pt")
13
  output = model.generate(
14
  input_ids,
15
+ max_length=300, # Maximum number of tokens
16
  temperature=0.9, # Adjust randomness
17
  top_k=50, # Top-k sampling
18
  top_p=0.95, # Nucleus sampling