foxxy-hm commited on
Commit
4450172
·
1 Parent(s): 9358218

Update src/models/qa_model.py

Browse files
Files changed (1) hide show
  1. src/models/qa_model.py +2 -2
src/models/qa_model.py CHANGED
@@ -2,8 +2,8 @@ import numpy as np
2
  import torch
3
  import torch.nn as nn
4
  from transformers import AutoModelForQuestionAnswering, pipeline
5
- from features.text_utils import post_process_answer
6
- from features.graph_utils import find_best_cluster
7
 
8
 
9
  class QAEnsembleModel(nn.Module):
 
2
  import torch
3
  import torch.nn as nn
4
  from transformers import AutoModelForQuestionAnswering, pipeline
5
+ from src.features.text_utils import post_process_answer
6
+ from src.features.graph_utils import find_best_cluster
7
 
8
 
9
  class QAEnsembleModel(nn.Module):