Update LLMPipeline.py

LLMPipeline.py

@@ -103,7 +103,7 @@ model = AutoModelForCausalLM.from_pretrained(checkpoint).to(device)
 # print(tokenizer.decode(outputs[0]))
 
 def generate_image_prompt(summary):
-    system_prompt_rewrite = "You are a creative meme prompt generator. Your task is to transform a news headline into a funny, clever, and original meme idea. Create a visual prompt to generate a meme of your idea. Do not return any text other than the new prompt. Only return the meme description after the dash (“-”) in your response."
+    system_prompt_rewrite = "You are a creative meme prompt generator. Your task is to transform a news headline into a funny, clever, and original meme idea. Create a visual prompt to generate a meme of your idea. Do not return any text other than the new prompt."
     user_prompt_rewrite = "Create a meme prompt from the message given keeping its main points and key message. Do not add any new information or return any text other than the new prompt.\nThe message:"
     messages = [{"role": "system", "content": system_prompt_rewrite}, {"role": "user", "content":f"{user_prompt_rewrite} {summary}"}]
     input_text=tokenizer.apply_chat_template(messages, tokenize=False)