Commit ·
47eb45b
1
Parent(s): 2c6600b
Update app.py
Browse files
app.py
CHANGED
|
@@ -27,7 +27,7 @@ def predict(question, lower_threshold, tags):
|
|
| 27 |
retrieved_doc_id = np.argmax(similarities)
|
| 28 |
|
| 29 |
if similarities[0][retrieved_doc_id] < float(lower_threshold):
|
| 30 |
-
ans = pipe(
|
| 31 |
ans['query_similarity_score'] = similarities[0][retrieved_doc_id]
|
| 32 |
return ans
|
| 33 |
return {"sequence" : data[retrieved_doc_id][-1], 'query_similarity_score' : similarities[0][retrieved_doc_id]}
|
|
|
|
| 27 |
retrieved_doc_id = np.argmax(similarities)
|
| 28 |
|
| 29 |
if similarities[0][retrieved_doc_id] < float(lower_threshold):
|
| 30 |
+
ans = pipe(question, candidate_labels=[x.strip() for x in tags.split(",") if x.strip()!=""])
|
| 31 |
ans['query_similarity_score'] = similarities[0][retrieved_doc_id]
|
| 32 |
return ans
|
| 33 |
return {"sequence" : data[retrieved_doc_id][-1], 'query_similarity_score' : similarities[0][retrieved_doc_id]}
|