aibmedia commited on
Commit
cdefac5
·
verified ·
1 Parent(s): 84c3c63

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +3 -1
main.py CHANGED
@@ -1,7 +1,9 @@
1
  import requests
 
2
 
3
  API_URL = "https://api-inference.huggingface.co/models/sentence-transformers/all-MiniLM-L6-v2"
4
- headers = {"Authorization": "Bearer "}
 
5
 
6
  from flask import Flask
7
  app = Flask(__name__)
 
1
  import requests
2
+ import os
3
 
4
  API_URL = "https://api-inference.huggingface.co/models/sentence-transformers/all-MiniLM-L6-v2"
5
+ bearer = "Bearer " + os.getenv('TOKEN')
6
+ headers = {"Authorization": bearer }
7
 
8
  from flask import Flask
9
  app = Flask(__name__)