import io from urllib.parse import urlparse import requests def load_file(path_or_url: str) -> io.BytesIO: if _is_url(path_or_url): response = requests.get(path_or_url) response.raise_for_status() return io.BytesIO(response.content) else: with open(path_or_url, 'rb') as f: return io.BytesIO(f.read()) def _is_url(path_or_url): try: result = urlparse(path_or_url) return result.scheme in ('http', 'https') except ValueError: return False