Commit ·
c748858
1
Parent(s): a78d024
Change 'sequence' name to 'chatbot_response'
Browse files
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 {"
|
| 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 |
|