Commit
·
5d1c407
1
Parent(s):
3247c3a
Handle multiple reutrn sequences
Browse files
app.py
CHANGED
|
@@ -173,6 +173,9 @@ def generate(input_message: str, db_info="", temperature=0.2, top_p=0.9, top_k=0
|
|
| 173 |
response_text = response_text.split("### Response:")[1].strip()
|
| 174 |
|
| 175 |
formatted_text = format(response_text) if format_sql else response_text
|
|
|
|
|
|
|
|
|
|
| 176 |
responses.append(formatted_text)
|
| 177 |
|
| 178 |
# Concat responses to be a single string seperated by a newline
|
|
|
|
| 173 |
response_text = response_text.split("### Response:")[1].strip()
|
| 174 |
|
| 175 |
formatted_text = format(response_text) if format_sql else response_text
|
| 176 |
+
if (num_return_sequences > 1):
|
| 177 |
+
formatted_text = formatted_text.replace("\n", "").replace("\t", "").strip()
|
| 178 |
+
|
| 179 |
responses.append(formatted_text)
|
| 180 |
|
| 181 |
# Concat responses to be a single string seperated by a newline
|