more text generated
Browse files
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=
|
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
|