yushihu commited on
Commit
d165671
1 Parent(s): d6776e0

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -5
README.md CHANGED
@@ -43,11 +43,7 @@ pipeline = transformers.pipeline(
43
 
44
  # formating prompt following LLaMA 2 style
45
  def create_qg_prompt(caption):
46
-
47
- INTRO_BLURB = """Given an image description, generate one or two multiple-choice questions that verifies if the image description is correct.
48
- Classify each concept into a type (object, human, animal, food, activity, attribute, counting, color, material, spatial, location, shape, other), and then generate a question for each type.
49
- """
50
-
51
  formated_prompt = f"<s>[INST] <<SYS>>\n{INTRO_BLURB}\n<</SYS>>\n\n"
52
  formated_prompt += f"Description: {caption} [/INST] Entities:"
53
  return formated_prompt
 
43
 
44
  # formating prompt following LLaMA 2 style
45
  def create_qg_prompt(caption):
46
+ INTRO_BLURB = "Given an image description, generate one or two multiple-choice questions that verifies if the image description is correct.\nClassify each concept into a type (object, human, animal, food, activity, attribute, counting, color, material, spatial, location, shape, other), and then generate a question for each type.\n"
 
 
 
 
47
  formated_prompt = f"<s>[INST] <<SYS>>\n{INTRO_BLURB}\n<</SYS>>\n\n"
48
  formated_prompt += f"Description: {caption} [/INST] Entities:"
49
  return formated_prompt