jonathanjordan21 commited on
Commit
c748858
1 Parent(s): a78d024

Change 'sequence' name to 'chatbot_response'

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -31,7 +31,7 @@ def predict(question, lower_threshold, tags, multi_label):
31
  ans = pipe(question, candidate_labels=[x.strip() for x in tags.split(",") if x.strip()!=""], multi_label=multi_label)
32
  ans['query_similarity_score'] = similarities[0][retrieved_doc_id]
33
  return ans
34
- return {"sequence" : data[retrieved_doc_id][-1], 'query_similarity_score' : similarities[0][retrieved_doc_id]}
35
 
36
 
37
 
 
31
  ans = pipe(question, candidate_labels=[x.strip() for x in tags.split(",") if x.strip()!=""], multi_label=multi_label)
32
  ans['query_similarity_score'] = similarities[0][retrieved_doc_id]
33
  return ans
34
+ return {"chatbot_response" : data[retrieved_doc_id][-1], 'query_similarity_score' : similarities[0][retrieved_doc_id]}
35
 
36
 
37