Update README.md

README.md

@@ -23,10 +23,10 @@ tuned_model = LlavaForConditionalGeneration.from_pretrained(
 )
 
 image_path = ""
-prompt = ""
+prompt = f"USER: <image>\n你是个表情包专家，仔细观察、理解图片中的想表达的感觉，把这个感觉转换成关键词。\nASSISTANT:"
 image = Image.open(image_path).convert("RGB")
 inputs = processor(text=prompt, images=[image], return_tensors="pt").to("cuda")
-tuned_generated_ids = tuned_model.generate(**inputs, max_new_tokens=MAX_LENGTH)
+generated_ids = tuned_model.generate(**inputs, max_new_tokens=MAX_LENGTH)
 generated_texts = processor.batch_decode(generated_ids, skip_special_tokens=True)
 ```