devve1 commited on
Commit
19a1340
1 Parent(s): 69d7f54

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -127,7 +127,7 @@ def main(query: str, client: QdrantClient, collection_name: str, llm, dense_mode
127
 
128
  seen_values = set()
129
  result_metadatas = "\n".join(
130
- f'{key}: {value}'
131
  for metadata in metadatas
132
  for key, value in metadata.items()
133
  if (value not in seen_values and not seen_values.add(value))
@@ -168,7 +168,7 @@ def main(query: str, client: QdrantClient, collection_name: str, llm, dense_mode
168
  ], stop=["</s>"], temperature=0.7, max_tokens=3000)
169
 
170
  answer = output['choices'][0]['message']['content']
171
- answer_with_metadatas = f"{answer}\n\n{result_metadatas}"
172
 
173
  print(f'OUTPUT: {output}')
174
  return answer, answer_with_metadatas
 
127
 
128
  seen_values = set()
129
  result_metadatas = "\n".join(
130
+ f'{value}'
131
  for metadata in metadatas
132
  for key, value in metadata.items()
133
  if (value not in seen_values and not seen_values.add(value))
 
168
  ], stop=["</s>"], temperature=0.7, max_tokens=3000)
169
 
170
  answer = output['choices'][0]['message']['content']
171
+ answer_with_metadatas = f"{answer}\n\nSource(s) :\n{result_metadatas}"
172
 
173
  print(f'OUTPUT: {output}')
174
  return answer, answer_with_metadatas