getModelUris.js 1.2 KB

12345678910111213141516171819202122232425262728293031
  1. "use strict";
  2. Object.defineProperty(exports, "__esModule", { value: true });
  3. function getModelUris(uri, defaultModelName) {
  4. var defaultManifestFilename = defaultModelName + "-weights_manifest.json";
  5. if (!uri) {
  6. return {
  7. modelBaseUri: '',
  8. manifestUri: defaultManifestFilename
  9. };
  10. }
  11. if (uri === '/') {
  12. return {
  13. modelBaseUri: '/',
  14. manifestUri: "/" + defaultManifestFilename
  15. };
  16. }
  17. var protocol = uri.startsWith('http://') ? 'http://' : uri.startsWith('https://') ? 'https://' : '';
  18. uri = uri.replace(protocol, '');
  19. var parts = uri.split('/').filter(function (s) { return s; });
  20. var manifestFile = uri.endsWith('.json')
  21. ? parts[parts.length - 1]
  22. : defaultManifestFilename;
  23. var modelBaseUri = protocol + (uri.endsWith('.json') ? parts.slice(0, parts.length - 1) : parts).join('/');
  24. modelBaseUri = uri.startsWith('/') ? "/" + modelBaseUri : modelBaseUri;
  25. return {
  26. modelBaseUri: modelBaseUri,
  27. manifestUri: modelBaseUri === '/' ? "/" + manifestFile : modelBaseUri + "/" + manifestFile
  28. };
  29. }
  30. exports.getModelUris = getModelUris;
  31. //# sourceMappingURL=getModelUris.js.map