getModelUris.js 1.1 KB

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