schuler commited on
Commit
c6e9b1e
·
verified ·
1 Parent(s): 75c74b0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -3
app.py CHANGED
@@ -48,8 +48,9 @@ def respond(
48
 
49
  for hmessage in messages:
50
  role = "<|assistant|>" if hmessage['role'] == 'assistant' else "<|user|>"
51
- prompt += f"\n{role}\n{hmessage['content']}\n<|end|>\n"
52
- # prompt += f"\n<|user|>\n{message}\n<|end|><|assistant|>\n"
 
53
 
54
  # """
55
  # Generate the response
@@ -69,7 +70,7 @@ def respond(
69
  # Extract the assistant's response
70
  result = generated_text[len(prompt):].strip()
71
  # """
72
- # result = message+':'+prompt
73
  except Exception as error:
74
  exc_type, exc_obj, exc_tb = sys.exc_info()
75
  fname = os.path.split(exc_tb.tb_frame.f_code.co_filename)[1]
 
48
 
49
  for hmessage in messages:
50
  role = "<|assistant|>" if hmessage['role'] == 'assistant' else "<|user|>"
51
+ prompt += f"{role}{hmessage['content']}<|end|>"
52
+ # prompt += f"<|user|>{message}<|end|><|assistant|>"
53
+ prompt += f"<|assistant|>"
54
 
55
  # """
56
  # Generate the response
 
70
  # Extract the assistant's response
71
  result = generated_text[len(prompt):].strip()
72
  # """
73
+ result = prompt +':'+result
74
  except Exception as error:
75
  exc_type, exc_obj, exc_tb = sys.exc_info()
76
  fname = os.path.split(exc_tb.tb_frame.f_code.co_filename)[1]