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