File size: 682 Bytes
202fdfa 8f68280 202fdfa |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 |
import os
import requests
class Endpoint:
def __init__(self):
self.config_path = "https://storage.googleapis.com/sfr-vision-language-research/LAVIS/projects/blip2/config.json"
self._url = None
@property
def url(self):
if self._url is None:
self._url = self.get_url()
return self._url
def get_url(self):
response = requests.get(self.config_path)
config = response.json()
return config["endpoint"]
def get_token():
token = os.environ.get("auth_token")
if token is None:
raise ValueError("auth-token not found in environment variables")
return token
|