Update app.py

app.py

@@ -17,6 +17,33 @@ from transformers import BlenderbotTokenizer, BlenderbotForConditionalGeneration
 from transformers import BlenderbotTokenizer, BlenderbotForConditionalGeneration
 
 
+###
+# Definition of different purspose prompts
+# https://huggingface.co/spaces/Chris4K/rlhf-arena/edit/main/app.py
+####
+def prompt_human_instruct(system_msg, history):
+    return system_msg.strip() + "\n" + \
+        "\n".join(["\n".join(["###Human: "+item[0], "###Assistant: "+item[1]])
+        for item in history])
+
+
+def prompt_instruct(system_msg, history):
+    return system_msg.strip() + "\n" + \
+        "\n".join(["\n".join(["### Instruction: "+item[0], "### Response: "+item[1]])
+        for item in history])
+
+
+def prompt_chat(system_msg, history):
+    return system_msg.strip() + "\n" + \
+        "\n".join(["\n".join(["USER: "+item[0], "ASSISTANT: "+item[1]])
+        for item in history])
+
+
+def prompt_roleplay(system_msg, history):
+    return "<|system|>" + system_msg.strip() + "\n" + \
+        "\n".join(["\n".join(["<|user|>"+item[0], "<|model|>"+item[1]])
+        for item in history])
+
 
 ####
 ## Sentinent models
@@ -45,17 +72,17 @@ chat_model_facebook_blenderbot_400M_distill = "facebook/blenderbot-400M-distill"
 chat_model_HenryJJ_vincua_13b = "HenryJJ/vincua-13b"
 
 # https://colab.research.google.com/drive/1hrS6_g14EcOD4ezwSGlGX2zxJegX5uNX#scrollTo=NUwUR9U7qkld
-llm_hf = HuggingFaceHub(
-    repo_id= chat_model_HenryJJ_vincua_13b,
-    model_kwargs={"temperature":0.9 }
-)
+#llm_hf = HuggingFaceHub(
+#    repo_id= chat_model_HenryJJ_vincua_13b,
+#    model_kwargs={"temperature":0.9 }
+#)
 
 
 
 text = "Why did the chicken cross the road?"
 
-output_question_1 = llm_hf(text)
-print(output_question_1)
+#output_question_1 = llm_hf(text)
+#print(output_question_1)