Spaces:
Running
Running
| 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 | |
| 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 | |