BenBranyon commited on
Commit
913e64b
·
verified ·
1 Parent(s): 7f99c81

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -66,12 +66,12 @@ def generate(
66
  repetition_penalty: float = 1.2,
67
  ) -> Iterator[str]:
68
  conversation = []
69
- system_prompt = "You are a rap lyric bot inspired by Sumkilla. Your lyrics promote liberation, dismantling oppression, and freedom, blending AI's role in uniting humanity and nature. Use humor, a unique voice, and rhyme as much as poosible, avoiding offensive words and slurs."
70
  if system_prompt:
71
  conversation.append({"role": "system", "content": system_prompt})
72
  for user, assistant in chat_history:
73
  conversation.extend([{"role": "user", "content": user}, {"role": "assistant", "content": assistant}])
74
- conversation.append({"role": "user", "content": "Generate rap lyircs about " + message + " using a similar style as the artist Sumkilla. Only generate rap lyircs and start each output with a song stucture like [VERSE 1]. Don't repeate your instructions in the output."})
75
 
76
  input_ids = tokenizer.apply_chat_template(conversation, return_tensors="pt")
77
  if input_ids.shape[1] > MAX_INPUT_TOKEN_LENGTH:
 
66
  repetition_penalty: float = 1.2,
67
  ) -> Iterator[str]:
68
  conversation = []
69
+ system_prompt = "You are a rap lyric bot inspired by Sumkilla. Your lyrics promote liberation, dismantling oppression, and freedom, blending AI's role in uniting humanity and nature. Use humor, a unique voice, and rhyme as much as poosible, avoiding offensive words and slurs. Only generate rap lyircs and start each output with a song stucture like [VERSE 1]."
70
  if system_prompt:
71
  conversation.append({"role": "system", "content": system_prompt})
72
  for user, assistant in chat_history:
73
  conversation.extend([{"role": "user", "content": user}, {"role": "assistant", "content": assistant}])
74
+ conversation.append({"role": "user", "content": "Generate rap lyircs using the style of the artist Sumkilla about " + message + ". Don't repeate your instructions in the output."})
75
 
76
  input_ids = tokenizer.apply_chat_template(conversation, return_tensors="pt")
77
  if input_ids.shape[1] > MAX_INPUT_TOKEN_LENGTH: