Upload folder using huggingface_hub

setup.sh

@@ -306,8 +306,10 @@ kv "Helpers" "screen -r helpers"
 section "RunPod restart"
 ok "After restart, run: bash ${INSTALL_BASE}/restart_after_reboot.sh"
 
-cp "${SCRIPT_DIR}/restart_after_reboot.sh" "${INSTALL_BASE}/restart_after_reboot.sh"
-chmod +x ${INSTALL_BASE}/restart_after_reboot.sh
+run_logged "download restart script" "${SETUP_LOG}" curl -fsSL \
+  -o "${INSTALL_BASE}/restart_after_reboot.sh" \
+  "https://huggingface.co/gtaayush010/comfysetup/resolve/main/restart_after_reboot.sh"
+chmod +x "${INSTALL_BASE}/restart_after_reboot.sh"
 
 section "Restart script saved"
 kv "Command" "bash ${INSTALL_BASE}/restart_after_reboot.sh"