alex6095 commited on
Commit
ed4e5a2
1 Parent(s): b6fd070

Fix No source message

Browse files
Files changed (2) hide show
  1. .gitattributes +1 -0
  2. app.py +3 -2
.gitattributes CHANGED
@@ -35,3 +35,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
35
  *tfevents* filter=lfs diff=lfs merge=lfs -text
36
  faiss_index/index.faiss filter=lfs diff=lfs merge=lfs -text
37
  parsed_texts.csv filter=lfs diff=lfs merge=lfs -text
 
 
35
  *tfevents* filter=lfs diff=lfs merge=lfs -text
36
  faiss_index/index.faiss filter=lfs diff=lfs merge=lfs -text
37
  parsed_texts.csv filter=lfs diff=lfs merge=lfs -text
38
+ faiss_index/index.pkl filter=lfs diff=lfs merge=lfs -text
app.py CHANGED
@@ -172,10 +172,11 @@ def generate_response(prompt):
172
  if src.page_content.startswith('Paragraph:'):
173
  sources.append(src.metadata['source'])
174
 
175
- if sources:
176
- message = message + "\n\nNo sources"
177
  else:
178
  message = message + "\n\n[" + ", ".join(sources) + "]"
 
179
  return message
180
 
181
 
 
172
  if src.page_content.startswith('Paragraph:'):
173
  sources.append(src.metadata['source'])
174
 
175
+ if len(sources)==0:
176
+ message = message + "\n\n[No sources]"
177
  else:
178
  message = message + "\n\n[" + ", ".join(sources) + "]"
179
+
180
  return message
181
 
182