helal94hb1 commited on
Commit
6868c13
·
1 Parent(s): 85c6371

feat: Update application with new changes6

Browse files
app/services/reranker_service.py CHANGED
@@ -6,7 +6,7 @@ import math
6
  import os
7
  # --- MODIFIED: Import your custom model architecture ---
8
  from app.models.expert_judge_model import ExpertJudgeCrossEncoder, get_tokenizer
9
-
10
  from app.core import state
11
  from app.core.config import settings
12
 
 
6
  import os
7
  # --- MODIFIED: Import your custom model architecture ---
8
  from app.models.expert_judge_model import ExpertJudgeCrossEncoder, get_tokenizer
9
+ import requests
10
  from app.core import state
11
  from app.core.config import settings
12
 
app/services/retrieval.py CHANGED
@@ -8,7 +8,7 @@ import torch.nn.functional as F
8
  from typing import List, Tuple
9
  from sentence_transformers import SentenceTransformer
10
  import os
11
-
12
  # Import settings and state
13
  from app.core.config import settings
14
  from app.core import state
 
8
  from typing import List, Tuple
9
  from sentence_transformers import SentenceTransformer
10
  import os
11
+ import requests
12
  # Import settings and state
13
  from app.core.config import settings
14
  from app.core import state