Update app.py
Browse files
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"
|
52 |
-
# prompt += f"
|
|
|
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 |
-
|
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]
|