refactor: use config yaml

app.py

@@ -1,21 +1,20 @@
 import streamlit as st
+import yaml
 import requests
 import re
 from pdfParser import get_pdf_text
 
+
 try:
     api_key = st.secrets.hf_credentials.hf_api
 except:
     api_key = st.secrets.hf_api
-model_id = "meta-llama/Llama-2-70b-chat-hf"
-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'}
-    """
+
+with open("config/model_config.yml", "r") as file:
+    model_config = yaml.safe_load(file)
+
+system_message = model_config["system_message"]
+model_id = model_config["model_id"]
 
 
 def query(payload, model_id):

config/model_config.yml

@@ -0,0 +1,2 @@
+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'}"
+model_id: meta-llama/Llama-2-70b-chat-hf