Update Dockerfile

Dockerfile

@@ -149,8 +149,12 @@ def restore():
 
     if not flag_exists and not force_restore:
         print("Restore skip: initialized.flag not found, this is first deploy or rebuild.")
-        client.put_file(INIT_FLAG, b'')
-        print(f"[OK] Uploaded {INIT_FLAG} to dataset")	
+        client = get_dataset_client()
+        try:
+            client.put_file(INIT_FLAG, b'')
+            print(f"[OK] Uploaded {INIT_FLAG} to dataset")
+        except Exception as e:
+            print(f"[Warn] Failed to upload {INIT_FLAG}: {e}")
         return
 
     if force_restore: