|
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 |
|
|
