Update app.py
Browse files
app.py
CHANGED
@@ -16,10 +16,7 @@ import os
|
|
16 |
HF_TOKEN = os.getenv('HF_Token')
|
17 |
#Log_Path="./Logfolder"
|
18 |
logfile = 'DiabetesChatLog.txt'
|
19 |
-
historylog = [
|
20 |
-
"Prompt": '',
|
21 |
-
"Output": ''
|
22 |
-
}]
|
23 |
|
24 |
data = load_dataset("Namitg02/Test", split='train', streaming=False)
|
25 |
#Returns a list of dictionaries, each representing a row in the dataset.
|
@@ -90,8 +87,12 @@ def format_prompt(prompt,retrieved_documents,k,history,memory_limit=3):
|
|
90 |
for user_message, bot_message in history[0:]:
|
91 |
PROMPT += f"<s>[INST] {user_message} [/INST] {bot_message} </s>"
|
92 |
print("checkwthhist2")
|
93 |
-
|
94 |
-
|
|
|
|
|
|
|
|
|
95 |
|
96 |
|
97 |
# Called by talk function to add retrieved documents to the prompt. Keeps adding text of retrieved documents to string that are retreived
|
@@ -104,6 +105,7 @@ def talk(prompt, history):
|
|
104 |
formatted_prompt = format_prompt(prompt,retrieved_documents,k,history,memory_limit=3) # create a new prompt using the retrieved documents
|
105 |
print("check5")
|
106 |
pd.options.display.max_colwidth = 4000
|
|
|
107 |
# print(retrieved_documents['0'])
|
108 |
# print(formatted_prompt)
|
109 |
# formatted_prompt_with_history = add_history(formatted_prompt, history)
|
|
|
16 |
HF_TOKEN = os.getenv('HF_Token')
|
17 |
#Log_Path="./Logfolder"
|
18 |
logfile = 'DiabetesChatLog.txt'
|
19 |
+
historylog = []
|
|
|
|
|
|
|
20 |
|
21 |
data = load_dataset("Namitg02/Test", split='train', streaming=False)
|
22 |
#Returns a list of dictionaries, each representing a row in the dataset.
|
|
|
87 |
for user_message, bot_message in history[0:]:
|
88 |
PROMPT += f"<s>[INST] {user_message} [/INST] {bot_message} </s>"
|
89 |
print("checkwthhist2")
|
90 |
+
|
91 |
+
for user_message, bot_message in history[0:]:
|
92 |
+
historylog += f"<s>[INST] {user_message} [/INST] {bot_message} </s>"
|
93 |
+
|
94 |
+
print(historylog)
|
95 |
+
return PROMPT, historylog
|
96 |
|
97 |
|
98 |
# Called by talk function to add retrieved documents to the prompt. Keeps adding text of retrieved documents to string that are retreived
|
|
|
105 |
formatted_prompt = format_prompt(prompt,retrieved_documents,k,history,memory_limit=3) # create a new prompt using the retrieved documents
|
106 |
print("check5")
|
107 |
pd.options.display.max_colwidth = 4000
|
108 |
+
print(historylog)
|
109 |
# print(retrieved_documents['0'])
|
110 |
# print(formatted_prompt)
|
111 |
# formatted_prompt_with_history = add_history(formatted_prompt, history)
|