Spaces:
Running
Running
File size: 3,484 Bytes
431b4de f8b5fef 081a4b3 f8b5fef 081a4b3 f8b5fef 081a4b3 f8b5fef 081a4b3 f8b5fef 081a4b3 431b4de 081a4b3 f8b5fef 081a4b3 f8b5fef 081a4b3 f8b5fef 081a4b3 431b4de |
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 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 |
const c = console;
const ENDPOINT = "https://huggingface.co";
async function whoami(token) {
const path = `${ENDPOINT}/api/whoami-v2`;
const res = await fetch(path, {
headers: {
Authorization: `Bearer ${token}`,
}
});
return await res.json();
}
/**
* @returns `str`: full URL to the newly created repo.
*/
async function createRepo(token, repoName, repoType) {
var _a;
const path = `${ENDPOINT}/api/repos/create`;
const res = await fetch(path, {
method: "POST",
headers: {
Authorization: `Bearer ${token}`,
"Content-Type": "application/json",
},
body: JSON.stringify({
name: repoName,
type: repoType,
})
});
const output = await res.json();
if (!res.ok) {
throw new Error(`Error ${res.status}: ${(_a = output.error) !== null && _a !== void 0 ? _a : output}`);
}
return output.url;
}
/**
* @returns `str`: URL to the newly uploaded blob.
*/
async function uploadFile(token, repoId, repoType, filename, blob) {
var _a;
const prefix = (repoType === "dataset") ? "/datasets"
: (repoType === "space") ? "/spaces"
: "";
const path = `${ENDPOINT}/api${prefix}/${repoId}/upload/main/${filename}`;
const res = await fetch(path, {
method: "POST",
headers: {
Authorization: `Bearer ${token}`,
},
body: blob
});
const output = await res.json();
if (!res.ok) {
throw new Error(`Error ${res.status}: ${(_a = output.error) !== null && _a !== void 0 ? _a : output}`);
}
return output.url;
}
const FILES_TO_UPLOAD = [
"./mobilenet/model.json",
"./mobilenet/group1-shard1of2",
"./mobilenet/group1-shard2of2",
"./mobilenet/coffee.jpg",
"./mobilenet/README.md",
];
function filenameFromURL(url) {
return url.substring(url.lastIndexOf("/") + 1);
}
window.addEventListener("load", function () {
const tokenEl = document.querySelector("#token");
const repoNameEl = document.querySelector("#repo_name");
const button = document.querySelector("#submit");
const output = document.querySelector("#logs");
const storedToken = window.localStorage.getItem("hf_token");
if (storedToken) {
tokenEl.value = storedToken;
/// ^to help in dev.
}
repoNameEl.value = `tfjs-mobilenet-${Date.now() % 1000}`;
/// "random" repo name
button.addEventListener("click", async function () {
const token = tokenEl.value;
const repoName = repoNameEl.value;
if (!token || !repoName) {
alert("You need a token and a repo name");
return;
}
button.setAttribute("disabled", "disabled");
try {
const fullUrl = await createRepo(token, repoName, "model");
const repoId = fullUrl.replace(ENDPOINT, "").replace(/^\//, "");
for (const file of FILES_TO_UPLOAD) {
const blob = await (await fetch(file)).blob();
await uploadFile(token, repoId, "model", filenameFromURL(file), blob);
}
button.insertAdjacentHTML("afterend", `<div class="text-green-500 mb-6">🎉 Upload complete! Model page is <a target="_blank" class="text-bold underline" href="${fullUrl}">${fullUrl}</a></div>`);
}
catch (err) {
output.append("\n" + err);
}
button.removeAttribute("disabled");
});
});
|