Namitg02 commited on
Commit
77b7a2a
·
verified ·
1 Parent(s): d2de33c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -5
app.py CHANGED
@@ -87,11 +87,11 @@ def format_prompt(prompt,retrieved_documents,k,history,memory_limit=3):
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
96
 
97
 
@@ -121,6 +121,7 @@ def talk(prompt, history):
121
  stream = model.create_chat_completion(messages = messages, max_tokens=1000, stop=["</s>"], stream=False)
122
  # print(f"{stream}")
123
  print("check 7")
 
124
  print(stream['choices'][0]['message']['content'])
125
  return(stream['choices'][0]['message']['content'])
126
  # text = ""
@@ -164,7 +165,7 @@ demo = gr.ChatInterface(
164
  )
165
  # launch chatbot and calls the talk function which in turn calls other functions
166
  print("check14")
167
- #print(historylog)
168
  #memory_panda = pd.DataFrame(historylog)
169
  #Logfile = Dataset.from_pandas(memory_panda)
170
  #Logfile.push_to_hub("Namitg02/Logfile",token = HF_TOKEN)
 
87
  for user_message, bot_message in history[0:]:
88
  PROMPT += f"<s>[INST] {user_message} [/INST] {bot_message} </s>"
89
  print("checkwthhist2")
90
+ global historylog
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
96
 
97
 
 
121
  stream = model.create_chat_completion(messages = messages, max_tokens=1000, stop=["</s>"], stream=False)
122
  # print(f"{stream}")
123
  print("check 7")
124
+ print(historylog)
125
  print(stream['choices'][0]['message']['content'])
126
  return(stream['choices'][0]['message']['content'])
127
  # text = ""
 
165
  )
166
  # launch chatbot and calls the talk function which in turn calls other functions
167
  print("check14")
168
+
169
  #memory_panda = pd.DataFrame(historylog)
170
  #Logfile = Dataset.from_pandas(memory_panda)
171
  #Logfile.push_to_hub("Namitg02/Logfile",token = HF_TOKEN)