Update README.md

README.md

@@ -171,46 +171,10 @@ generation_config = dict(
 )
 
 # single-round single-image conversation
-question = "请详细描述图片" # Please describe the picture in detail
+question = "Describe this image in detail" # Please describe the picture in detail
 response = model.chat(tokenizer, pixel_values, question, generation_config)
 print(question, response)
 
-# multi-round single-image conversation
-question = "请详细描述图片" # Please describe the picture in detail
-response, history = model.chat(tokenizer, pixel_values, question, generation_config, history=None, return_history=True)
-print(question, response)
-
-question = "请根据图片写一首诗" # Please write a poem according to the picture
-response, history = model.chat(tokenizer, pixel_values, question, generation_config, history=history, return_history=True)
-print(question, response)
-
-# multi-round multi-image conversation
-pixel_values1 = load_image('./examples/image1.jpg', max_num=6).to(torch.bfloat16).cuda()
-pixel_values2 = load_image('./examples/image2.jpg', max_num=6).to(torch.bfloat16).cuda()
-pixel_values = torch.cat((pixel_values1, pixel_values2), dim=0)
-
-question = "详细描述这两张图片" # Describe the two pictures in detail
-response, history = model.chat(tokenizer, pixel_values, question, generation_config, history=None, return_history=True)
-print(question, response)
-
-question = "这两张图片的相同点和区别分别是什么" # What are the similarities and differences between these two pictures
-response, history = model.chat(tokenizer, pixel_values, question, generation_config, history=history, return_history=True)
-print(question, response)
-
-# batch inference (single image per sample)
-pixel_values1 = load_image('./examples/image1.jpg', max_num=6).to(torch.bfloat16).cuda()
-pixel_values2 = load_image('./examples/image2.jpg', max_num=6).to(torch.bfloat16).cuda()
-image_counts = [pixel_values1.size(0), pixel_values2.size(0)]
-pixel_values = torch.cat((pixel_values1, pixel_values2), dim=0)
-
-questions = ["Describe the image in detail."] * len(image_counts)
-responses = model.batch_chat(tokenizer, pixel_values,
-                             image_counts=image_counts,
-                             questions=questions,
-                             generation_config=generation_config)
-for question, response in zip(questions, responses):
-    print(question)
-    print(response)
 ```
 
 ## Citation