jonathanjordan21
commited on
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 |
|