luiscgp commited on
Commit
06a6b06
1 Parent(s): 8e95425
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -40,8 +40,8 @@ class EntailmentChecker(BaseComponent):
40
  tokenizer: Optional[str] = None,
41
  use_gpu: bool = True,
42
  batch_size: int = 100,
43
- entailment_contradiction_consideration: float = 0.6,
44
- entailment_contradiction_threshold: float = 0.8
45
  ):
46
  """
47
  Load a Natural Language Inference model from Transformers.
 
40
  tokenizer: Optional[str] = None,
41
  use_gpu: bool = True,
42
  batch_size: int = 100,
43
+ entailment_contradiction_consideration: float = 0.7,
44
+ entailment_contradiction_threshold: float = 0.95
45
  ):
46
  """
47
  Load a Natural Language Inference model from Transformers.