Tonic commited on
Commit
e64092e
1 Parent(s): 42c77e0

fix sentence transformer latest version

Browse files
Files changed (2) hide show
  1. app.py +2 -0
  2. requirements.txt +2 -2
app.py CHANGED
@@ -20,6 +20,8 @@ import torch.nn.functional as F
20
  from dotenv import load_dotenv
21
  from utils import load_env_variables, parse_and_route, escape_special_characters
22
  from globalvars import API_BASE, intention_prompt, tasks, system_message, model_name, metadata_prompt
 
 
23
 
24
  load_dotenv()
25
 
 
20
  from dotenv import load_dotenv
21
  from utils import load_env_variables, parse_and_route, escape_special_characters
22
  from globalvars import API_BASE, intention_prompt, tasks, system_message, model_name, metadata_prompt
23
+ from sentence_transformers import SentenceTransformer
24
+
25
 
26
  load_dotenv()
27
 
requirements.txt CHANGED
@@ -1,6 +1,6 @@
1
  huggingface_hub
2
  einops
3
- sentence-transformers==2.2.2
4
  torch==2.2.0
5
  transformers
6
  openai
@@ -15,4 +15,4 @@ gradio
15
  # tesseract
16
  # libxml2
17
  # libxslt
18
- InstructorEmbedding==1.0.1
 
1
  huggingface_hub
2
  einops
3
+ sentence-transformers
4
  torch==2.2.0
5
  transformers
6
  openai
 
15
  # tesseract
16
  # libxml2
17
  # libxslt
18
+ git+https://github.com/xlang-ai/instructor-embedding.git@4721e7375afeb8fcb32400a13057f9348bb69392#egg=instructor-embedding