regraded01 commited on
Commit
5976640
1 Parent(s): dc68d68

feat: include prompt template

Browse files
Files changed (1) hide show
  1. config/model_config.yml +2 -1
config/model_config.yml CHANGED
@@ -1,2 +1,3 @@
1
  system_message: "Your role is to take PDF documents and extract their raw text into a JSON format that can be uploaded into a database. Return the JSON only. For example if you need to extract information about a report written on 2nd February 2011 with an author called Jane Mary then return this only: {'report_written_date': '02/02/2011', 'author_name': 'Jane Mary'} Another example would be a clinical exam passed by a student on the 3rd of July 2022 would return this only: {'result' : 'pass', 'date_of_exam' : '03/07/2022'}"
2
- model_id: meta-llama/Llama-2-70b-chat-hf
 
 
1
  system_message: "Your role is to take PDF documents and extract their raw text into a JSON format that can be uploaded into a database. Return the JSON only. For example if you need to extract information about a report written on 2nd February 2011 with an author called Jane Mary then return this only: {'report_written_date': '02/02/2011', 'author_name': 'Jane Mary'} Another example would be a clinical exam passed by a student on the 3rd of July 2022 would return this only: {'result' : 'pass', 'date_of_exam' : '03/07/2022'}"
2
+ prompt_template_string: "System: {system_message}\nUser: {user_message}"
3
+ model_id: "meta-llama/Llama-2-70b-chat-hf"