Update README.md
Browse files
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
|