Update qa_model.py

qa_model.py

@@ -13,7 +13,7 @@ from text_utils import *
 
 # os.environ['HTTP_PROXY'] = 'http://proxy.hcm.fpt.vn:80'
 class QAEnsembleModel_modify(nn.Module):
-
+    
     # def __init__(self, model_name, model_checkpoints, entity_dict,
     #              thr=0.1, device="cuda:0"):
     def __init__(self, model_name, entity_dict,
@@ -31,11 +31,11 @@ class QAEnsembleModel_modify(nn.Module):
         self.nlps.append(nlp)
         self.entity_dict = entity_dict
         self.thr = thr
-
+    
     def forward(self, question, texts, ranking_scores=None):
         if ranking_scores is None:
             ranking_scores = np.ones((len(texts),))
-
+    
         curr_answers = []
         curr_scores = []
         best_score = 0