DevNumb commited on
Commit
02875da
·
verified ·
1 Parent(s): 5d6a961

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -5,10 +5,10 @@ import docx
5
  import requests
6
  import numpy as np
7
 
8
- HF_MODEL = "sentence-transformers/all-MiniLM-L6-v2"
9
- HF_API_URL = f"https://api-inference.huggingface.co/models/{HF_MODEL}"
10
- HF_TOKEN = os.getenv("HF_TOKEN") # set this secret in your Space
11
- HEADERS = {"Authorization": f"Bearer {HF_TOKEN}"}
12
 
13
 
14
  # ---- Text extraction ----
 
5
  import requests
6
  import numpy as np
7
 
8
+ API_URL = "https://router.huggingface.co/hf-inference/models/sentence-transformers/all-MiniLM-L6-v2/pipeline/sentence-similarity"
9
+ headers = {
10
+ "Authorization": f"Bearer {os.environ['HF_TOKEN']}",
11
+ }
12
 
13
 
14
  # ---- Text extraction ----