Jorge Henao commited on
Commit
a5df69d
β€’
1 Parent(s): b480d8c
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -27,7 +27,7 @@ class DocumentQueries(ABC):
27
 
28
  class ExtractiveProposalQueries(DocumentQueries):
29
 
30
- def __init__(self, es_host: str, es_index: str, es_user, es_password, reader_name_or_path: str, use_gpu = False) -> None:
31
  reader = FARMReader(model_name_or_path = reader_name_or_path, use_gpu = use_gpu, num_processes=1)
32
  self._initialize_pipeline(es_host, es_index, es_user, es_password, reader = reader)
33
 
 
27
 
28
  class ExtractiveProposalQueries(DocumentQueries):
29
 
30
+ def __init__(self, es_host: str, es_index: str, es_user, es_password, reader_name_or_path: str, use_gpu = True) -> None:
31
  reader = FARMReader(model_name_or_path = reader_name_or_path, use_gpu = use_gpu, num_processes=1)
32
  self._initialize_pipeline(es_host, es_index, es_user, es_password, reader = reader)
33