Spaces:
Sleeping
Sleeping
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
|