SauleBis commited on
Commit
ecb5b52
1 Parent(s): 53b90ec

update req

Browse files
Files changed (2) hide show
  1. app.py +1 -1
  2. requirements.txt +56 -0
app.py CHANGED
@@ -35,7 +35,7 @@ def embed_bert_cls(text, model=model, tokenizer=tokenizer):
35
  return embeddings[0].cpu().squeeze()
36
 
37
  # Load embeddings
38
- embeddings = np.loadtxt('models/embeddings.txt')
39
  embeddings_tensor = [torch.tensor(embedding) for embedding in embeddings]
40
 
41
  # Create Faiss index
 
35
  return embeddings[0].cpu().squeeze()
36
 
37
  # Load embeddings
38
+ embeddings = np.loadtxt('embeddings.txt')
39
  embeddings_tensor = [torch.tensor(embedding) for embedding in embeddings]
40
 
41
  # Create Faiss index
requirements.txt CHANGED
@@ -0,0 +1,56 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ altair==5.3.0
2
+ attrs==23.2.0
3
+ blinker==1.8.2
4
+ cachetools==5.3.3
5
+ certifi==2024.7.4
6
+ charset-normalizer==3.3.2
7
+ click==8.1.7
8
+ faiss-cpu==1.8.0.post1
9
+ filelock==3.15.4
10
+ fsspec==2024.6.1
11
+ gitdb==4.0.11
12
+ GitPython==3.1.43
13
+ huggingface-hub==0.23.4
14
+ idna==3.7
15
+ Jinja2==3.1.4
16
+ jsonschema==4.23.0
17
+ jsonschema-specifications==2023.12.1
18
+ markdown-it-py==3.0.0
19
+ MarkupSafe==2.1.5
20
+ mdurl==0.1.2
21
+ mpmath==1.3.0
22
+ networkx==3.3
23
+ numpy==1.26.4
24
+ packaging==24.1
25
+ pandas==2.2.2
26
+ pillow==10.4.0
27
+ protobuf==5.27.2
28
+ pyarrow==16.1.0
29
+ pydeck==0.9.1
30
+ Pygments==2.18.0
31
+ python-dateutil==2.9.0.post0
32
+ pytz==2024.1
33
+ PyYAML==6.0.1
34
+ referencing==0.35.1
35
+ regex==2024.5.15
36
+ requests==2.32.3
37
+ rich==13.7.1
38
+ rpds-py==0.19.0
39
+ safetensors==0.4.3
40
+ six==1.16.0
41
+ smmap==5.0.1
42
+ streamlit==1.36.0
43
+ sympy==1.13.0
44
+ tenacity==8.5.0
45
+ tokenizers==0.19.1
46
+ toml==0.10.2
47
+ toolz==0.12.1
48
+ torch==2.3.1
49
+ torchaudio==2.3.1
50
+ torchvision==0.18.1
51
+ tornado==6.4.1
52
+ tqdm==4.66.4
53
+ transformers==4.42.4
54
+ typing_extensions==4.12.2
55
+ tzdata==2024.1
56
+ urllib3==2.2.2