Create README.md

README.md

@@ -0,0 +1,28 @@
+```
+def wrap_prompt(prompt, input):
+  if model_name == "HuggingFaceH4/zephyr-7b-alpha":
+    return f"### Instructions:\n{prompt}" + "\n---\n" + input + "\n\n### Response: "
+  elif model_name == "stabilityai/stablelm-3b-4e1t":
+    return f"{prompt}" + "Input: " + input + "\n Output: "
+  # elif model_name == "princeton-nlp/Sheared-LLaMA-1.3B":
+  elif "LLaMA" in model_name:
+    return f"<s>[INST] <<SYS>>\n{prompt}\n<</SYS>>\n\n{input} [/INST] "
+  else:
+    raise ValueError("Unrecognised model")
+
+
+def add_prompt(input: str):
+  prompt = '\n'.join(["You prepare students for exams by turning some text they provide into flashcards.",
+                  "The flashcards must be statements and not question & answers",
+                  "You try to represent everything in the text but ignore information that wouldn't be useful in an exam e.g. bibliographies, references, advertisements.",
+                  "Each piece of information should be in the same language that it was written in the text. Do not translate it to english.",
+                  "The flashcards should be independent, concise, formulated in a single sentence, unambiguous and factual.",
+                  "The flashcards should make sense on their own without any additional context e.g. they shouldn't use words like 'he', 'it', 'they' unless it's clear what they're referring to",
+                  "The flashcards should only contain knowledge found directly in the text. Do not include information from other sources",
+                  "Copy the formatting style they use for the text, e.g. if the text is a list of Term: Definition inputs then keep the same Term: Definition format",
+                  "Write out any equations exactly as they appear in the text, do not remove maths symbols like '+' or '='",
+                  f"Before each flashcard write {start_item_icon} and after each flashcard write {end_item_icon}",
+                    ])
+  return wrap_prompt(prompt, input)
+
+```