File size: 2,033 Bytes
894fa47 | 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 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 | import os
import requests
from tqdm import tqdm
from requests_toolbelt.multipart.encoder import MultipartEncoder, MultipartEncoderMonitor
URL = r"https://yvkmaiiycqligmloxwbx.supabase.co"
KEY="eyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCJ9.eyJpc3MiOiJzdXBhYmFzZSIsInJlZiI6Inl2a21haWl5Y3FsaWdtbG94d2J4Iiwicm9sZSI6InNlcnZpY2Vfcm9sZSIsImlhdCI6MTc3MDQ1MDQ2NCwiZXhwIjoyMDg2MDI2NDY0fQ.gxBw1bzwDuA1tTwDAoekHArpvhzNHIuOdZxJaPy7lnI"
BUCKET = "diagnostic-model"
LOCAL_FILE = r"D:\Projects\CytoSight\cytosight\backend\app\phikonv2\model.safetensors"
REMOTE_PATH = "v1/model.safetensors"
def upload_with_fast_failure():
file_size = os.path.getsize(LOCAL_FILE)
upload_url = f"{URL}/storage/v1/object/{BUCKET}/{REMOTE_PATH}"
print(f"🚀 Preparing Upload: {os.path.basename(LOCAL_FILE)} ({file_size / (1024**3):.2f} GB)")
with open(LOCAL_FILE, 'rb') as f:
pbar = tqdm(total=file_size, unit='B', unit_scale=True, desc="📡 Uploading")
def callback(monitor):
pbar.update(monitor.bytes_read - pbar.n)
encoder = MultipartEncoder(fields={'file': (os.path.basename(LOCAL_FILE), f, 'application/octet-stream')})
monitor = MultipartEncoderMonitor(encoder, callback)
headers = {
'Authorization': f'Bearer {KEY}',
'Content-Type': monitor.content_type,
'x-upsert': 'true',
'Expect': '100-continue'
}
try:
session = requests.Session()
response = session.post(upload_url, data=monitor, headers=headers, timeout=None)
pbar.close()
if response.status_code in [200, 201]:
print("\n✅ Upload successful!")
else:
print(f"\n❌ Server Rejected Upload: {response.status_code}")
print(f"Message: {response.text}")
except requests.exceptions.RequestException as e:
pbar.close()
print(f"\n❌ Network Error: {e}")
if __name__ == "__main__":
upload_with_fast_failure() |