Update app.py

app.py

@@ -37,7 +37,7 @@ def generate_song(prompt_text = ''):
         input_template = tokenizer.apply_chat_template([{"role": "user", "content": prompt_text}], tokenize=False, add_generation_prompt=True)
         input_ids = tokenizer(input_template, return_tensors="pt").to(model.device)
         #sample_outputs = model.generate(**input_ids, max_new_tokens=512 , repetition_penalty=1.1, temperature=0.4, top_p=0.95, top_k=40, do_sample = True)
-        sample_outputs = model.generate(**input_ids, max_new_tokens=256 , repetition_penalty=1.1, temperature=0.6, top_p=0.4, top_k=40, do_sample = True)
+        sample_outputs = model.generate(**input_ids, max_new_tokens=384 , repetition_penalty=1.1, temperature=0.6, top_p=0.4, top_k=40, do_sample = True)
         #sample_outputs = model.generate(**input_ids, max_new_tokens=512 , repetition_penalty=1.1, temperature=0.5, do_sample = True)
         decoded_output = tokenizer.batch_decode(sample_outputs, skip_special_tokens=True)[0]
         result = decoded_output.replace("user\n", "משתמש:\n").replace("model\n", "\nמודל:\n")
@@ -52,7 +52,7 @@ demo = gr.Interface(
     title=title,
     description=DESCRIPTION,
     article=article,
-    examples=["תנו לשמש לעלות, לבוקר להאיר", "כתוב לי בבקשה שיר על תפוח אדמה עם חרדה חברתית", "שיר המתאר את חייהם של חזירים בצבע סגול עם כנפיים וההרפתקאות המעופפות שלהם", "שיר על פתח תקווה"],
+    examples=["תנו לשמש לעלות, לבוקר להאיר", "כתוב לי בבקשה שיר על תפוח אדמה עם חרדה חברתית", "שיר המתאר את חייהם של חזירים בצבע סגול עם כנפיים וההרפתקאות המעופפות שלהם", "שיר על פתח תקווה", "שיר על חדי קרן וורודים 🦄"],
     allow_flagging="never",
 )