IotaCluster commited on
Commit
0cf4c45
·
verified ·
1 Parent(s): 18c9386

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +16 -4
requirements.txt CHANGED
@@ -1,4 +1,16 @@
1
- gradio
2
- sentence-transformers
3
- transformers
4
- torch
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # Gradio UI
2
+ gradio>=3.0
3
+
4
+ # Dense embeddings
5
+ sentence-transformers>=2.2.2
6
+ torch>=1.12.0
7
+
8
+ # Sparse BM25
9
+ rank_bm25>=0.2.2
10
+
11
+ # Late-interaction (ColBERT)
12
+ transformers>=4.26.0
13
+ colbert-py>=1.0.0 # or replace with your ColBERT package name/version
14
+
15
+ # (Optional) HF hub support
16
+ huggingface-hub>=0.13.0