Initial codes commit

models/reranker.py

@@ -21,7 +21,7 @@ def rerank(request: Request, query: str, contexts: List[Dict]) -> List[Dict]:
     pairs = [(query, ctx["text"]) for ctx in contexts]
     inputs = tokenizer(pairs, return_tensors="np", padding=True, truncation=True, max_length=256)
     ort_inputs = {k: v for k, v in inputs.items()}
-    scores = sess.run(None, ort_inputs)[0]  # [batch, 1] 형태라고 가정
+    scores = sess.run(None, ort_inputs)[0]  # [batch, 1] 형태
     scores = scores.squeeze(-1)
 
     for ctx, sc in zip(contexts, scores):

requirements.txt

@@ -14,12 +14,12 @@ pydantic==2.11.5
 # Hugging Face 관련
 huggingface_hub>=1.0.0
 datasets==3.0.1
-transformers>=4.44.0   # AutoTokenizer 사용을 위해 필요
+transformers>=4.44.0   
 
 # ONNX Runtime (모델 추론)
 onnxruntime>=1.19.0
 
 # 기타 유틸
-tqdm==4.66.5   # 데이터셋 처리 시 진행률 표시가 필요하다면
-requests>=2.32 # huggingface_hub 내부에서 사용되므로 최신 버전 권장
+tqdm==4.66.5  
+requests>=2.32