EYEDOL commited on
Commit
db07a3a
·
verified ·
1 Parent(s): 62b5323

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -5
app.py CHANGED
@@ -211,9 +211,8 @@ class WeeboAssistant:
211
  return output_path
212
 
213
  def get_llm_response(self, chat_history):
214
- prompt_lines = [
215
- self.SYSTEM_PROMPT.strip(),
216
- "" # Empty string or add intended string content here
217
  ]
218
 
219
 
@@ -224,8 +223,7 @@ class WeeboAssistant:
224
  if assistant_msg:
225
  prompt_lines.append("Assistant: " + assistant_msg)
226
  prompt_lines.append("Assistant: ")
227
- prompt = "
228
- ".join(prompt_lines)
229
 
230
  inputs = self.llm_tokenizer(prompt, return_tensors="pt")
231
  try:
 
211
  return output_path
212
 
213
  def get_llm_response(self, chat_history):
214
+ prompt_lines = [self.SYSTEM_PROMPT.strip(),
215
+ "" # Empty string or add intended string content here
 
216
  ]
217
 
218
 
 
223
  if assistant_msg:
224
  prompt_lines.append("Assistant: " + assistant_msg)
225
  prompt_lines.append("Assistant: ")
226
+ prompt = "".join(prompt_lines)
 
227
 
228
  inputs = self.llm_tokenizer(prompt, return_tensors="pt")
229
  try: