Spaces:
Paused
Paused
Davide Fiocco
commited on
Commit
·
abbccac
1
Parent(s):
4069b8e
Lower threshold for unsure
Browse files
utils.py
CHANGED
|
@@ -8,7 +8,7 @@ import torch
|
|
| 8 |
from transformers import Pipeline, pipeline
|
| 9 |
|
| 10 |
|
| 11 |
-
def get_answer(input, context, engine, threshold=0.
|
| 12 |
|
| 13 |
answer = engine({"question": input, "context": context})
|
| 14 |
|
|
|
|
| 8 |
from transformers import Pipeline, pipeline
|
| 9 |
|
| 10 |
|
| 11 |
+
def get_answer(input, context, engine, threshold=0.4):
|
| 12 |
|
| 13 |
answer = engine({"question": input, "context": context})
|
| 14 |
|