Update app.py

app.py

@@ -66,12 +66,12 @@ def generate(
     repetition_penalty: float = 1.2,
 ) -> Iterator[str]:
     conversation = []
-    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."
+    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."
     if system_prompt:
         conversation.append({"role": "system", "content": system_prompt})
     for user, assistant in chat_history:
         conversation.extend([{"role": "user", "content": user}, {"role": "assistant", "content": assistant}])
-    conversation.append({"role": "user", "content": "Generate rap lyircs about " + message + " using a similar style as the artist Sumkilla. Use humor, a unique voice, and rhyme as much as poosible, avoiding offensive words and slurs. Only print song lyircs and start each output with a song stucture like [VERSE 1]. Don't repeate your instructions in the output."})
+    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."})
 
     input_ids = tokenizer.apply_chat_template(conversation, return_tensors="pt")
     if input_ids.shape[1] > MAX_INPUT_TOKEN_LENGTH:
@@ -113,11 +113,11 @@ demo = gr.ChatInterface(
             step=1,
             value=DEFAULT_MAX_NEW_TOKENS,
         ),
-        gr.Slider(minimum=0.1, maximum=0.8, value=0.8, step=0.1, label="Temperature"),
+        gr.Slider(minimum=0.1, maximum=0.7, value=0.8, step=0.1, label="Temperature"),
         gr.Slider(
             minimum=0.1,
             maximum=1.0,
-            value=0.3,
+            value=0.2,
             step=0.05,
             label="Top-p (nucleus sampling)",
         ),