@huggingface/tasks 0.13.0-test3 → 0.13.0-test5

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (270) hide show
  1. package/dist/default-widget-inputs.d.cts +2 -2
  2. package/dist/default-widget-inputs.d.cts.map +1 -1
  3. package/dist/default-widget-inputs.d.ts +2 -2
  4. package/dist/default-widget-inputs.d.ts.map +1 -1
  5. package/dist/index.d.cts +19 -19
  6. package/dist/index.d.cts.map +1 -1
  7. package/dist/index.d.ts +19 -19
  8. package/dist/index.d.ts.map +1 -1
  9. package/dist/library-to-tasks.d.cts +2 -2
  10. package/dist/library-to-tasks.d.cts.map +1 -1
  11. package/dist/library-to-tasks.d.ts +2 -2
  12. package/dist/library-to-tasks.d.ts.map +1 -1
  13. package/dist/local-apps.d.cts +2 -2
  14. package/dist/local-apps.d.cts.map +1 -1
  15. package/dist/local-apps.d.ts +2 -2
  16. package/dist/local-apps.d.ts.map +1 -1
  17. package/dist/model-data.d.cts +3 -3
  18. package/dist/model-data.d.cts.map +1 -1
  19. package/dist/model-data.d.ts +3 -3
  20. package/dist/model-data.d.ts.map +1 -1
  21. package/dist/model-libraries-snippets.d.cts +1 -1
  22. package/dist/model-libraries-snippets.d.cts.map +1 -1
  23. package/dist/model-libraries-snippets.d.ts +1 -1
  24. package/dist/model-libraries-snippets.d.ts.map +1 -1
  25. package/dist/model-libraries.d.cts +2 -2
  26. package/dist/model-libraries.d.cts.map +1 -1
  27. package/dist/model-libraries.d.ts +2 -2
  28. package/dist/model-libraries.d.ts.map +1 -1
  29. package/dist/snippets/common.d.cts +1 -1
  30. package/dist/snippets/common.d.cts.map +1 -1
  31. package/dist/snippets/common.d.ts +1 -1
  32. package/dist/snippets/common.d.ts.map +1 -1
  33. package/dist/snippets/index.d.cts +5 -4
  34. package/dist/snippets/index.d.cts.map +1 -1
  35. package/dist/snippets/index.d.ts +5 -4
  36. package/dist/snippets/index.d.ts.map +1 -1
  37. package/dist/snippets/inputs.d.cts +2 -2
  38. package/dist/snippets/inputs.d.cts.map +1 -1
  39. package/dist/snippets/inputs.d.ts +2 -2
  40. package/dist/snippets/inputs.d.ts.map +1 -1
  41. package/dist/snippets/types.d.cts +1 -1
  42. package/dist/snippets/types.d.cts.map +1 -1
  43. package/dist/snippets/types.d.ts +1 -1
  44. package/dist/snippets/types.d.ts.map +1 -1
  45. package/dist/tasks/audio-classification/data.d.cts +1 -1
  46. package/dist/tasks/audio-classification/data.d.cts.map +1 -1
  47. package/dist/tasks/audio-classification/data.d.ts +1 -1
  48. package/dist/tasks/audio-classification/data.d.ts.map +1 -1
  49. package/dist/tasks/audio-to-audio/data.d.cts +1 -1
  50. package/dist/tasks/audio-to-audio/data.d.cts.map +1 -1
  51. package/dist/tasks/audio-to-audio/data.d.ts +1 -1
  52. package/dist/tasks/audio-to-audio/data.d.ts.map +1 -1
  53. package/dist/tasks/automatic-speech-recognition/data.d.cts +1 -1
  54. package/dist/tasks/automatic-speech-recognition/data.d.cts.map +1 -1
  55. package/dist/tasks/automatic-speech-recognition/data.d.ts +1 -1
  56. package/dist/tasks/automatic-speech-recognition/data.d.ts.map +1 -1
  57. package/dist/tasks/depth-estimation/data.d.cts +1 -1
  58. package/dist/tasks/depth-estimation/data.d.cts.map +1 -1
  59. package/dist/tasks/depth-estimation/data.d.ts +1 -1
  60. package/dist/tasks/depth-estimation/data.d.ts.map +1 -1
  61. package/dist/tasks/document-question-answering/data.d.cts +1 -1
  62. package/dist/tasks/document-question-answering/data.d.cts.map +1 -1
  63. package/dist/tasks/document-question-answering/data.d.ts +1 -1
  64. package/dist/tasks/document-question-answering/data.d.ts.map +1 -1
  65. package/dist/tasks/feature-extraction/data.d.cts +1 -1
  66. package/dist/tasks/feature-extraction/data.d.cts.map +1 -1
  67. package/dist/tasks/feature-extraction/data.d.ts +1 -1
  68. package/dist/tasks/feature-extraction/data.d.ts.map +1 -1
  69. package/dist/tasks/fill-mask/data.d.cts +1 -1
  70. package/dist/tasks/fill-mask/data.d.cts.map +1 -1
  71. package/dist/tasks/fill-mask/data.d.ts +1 -1
  72. package/dist/tasks/fill-mask/data.d.ts.map +1 -1
  73. package/dist/tasks/image-classification/data.d.cts +1 -1
  74. package/dist/tasks/image-classification/data.d.cts.map +1 -1
  75. package/dist/tasks/image-classification/data.d.ts +1 -1
  76. package/dist/tasks/image-classification/data.d.ts.map +1 -1
  77. package/dist/tasks/image-feature-extraction/data.d.cts +1 -1
  78. package/dist/tasks/image-feature-extraction/data.d.cts.map +1 -1
  79. package/dist/tasks/image-feature-extraction/data.d.ts +1 -1
  80. package/dist/tasks/image-feature-extraction/data.d.ts.map +1 -1
  81. package/dist/tasks/image-segmentation/data.d.cts +1 -1
  82. package/dist/tasks/image-segmentation/data.d.cts.map +1 -1
  83. package/dist/tasks/image-segmentation/data.d.ts +1 -1
  84. package/dist/tasks/image-segmentation/data.d.ts.map +1 -1
  85. package/dist/tasks/image-text-to-text/data.d.cts +1 -1
  86. package/dist/tasks/image-text-to-text/data.d.cts.map +1 -1
  87. package/dist/tasks/image-text-to-text/data.d.ts +1 -1
  88. package/dist/tasks/image-text-to-text/data.d.ts.map +1 -1
  89. package/dist/tasks/image-to-3d/data.d.cts +1 -1
  90. package/dist/tasks/image-to-3d/data.d.cts.map +1 -1
  91. package/dist/tasks/image-to-3d/data.d.ts +1 -1
  92. package/dist/tasks/image-to-3d/data.d.ts.map +1 -1
  93. package/dist/tasks/image-to-image/data.d.cts +1 -1
  94. package/dist/tasks/image-to-image/data.d.cts.map +1 -1
  95. package/dist/tasks/image-to-image/data.d.ts +1 -1
  96. package/dist/tasks/image-to-image/data.d.ts.map +1 -1
  97. package/dist/tasks/image-to-text/data.d.cts +1 -1
  98. package/dist/tasks/image-to-text/data.d.cts.map +1 -1
  99. package/dist/tasks/image-to-text/data.d.ts +1 -1
  100. package/dist/tasks/image-to-text/data.d.ts.map +1 -1
  101. package/dist/tasks/index.d.cts +29 -29
  102. package/dist/tasks/index.d.cts.map +1 -1
  103. package/dist/tasks/index.d.ts +29 -29
  104. package/dist/tasks/index.d.ts.map +1 -1
  105. package/dist/tasks/keypoint-detection/data.d.cts +1 -1
  106. package/dist/tasks/keypoint-detection/data.d.cts.map +1 -1
  107. package/dist/tasks/keypoint-detection/data.d.ts +1 -1
  108. package/dist/tasks/keypoint-detection/data.d.ts.map +1 -1
  109. package/dist/tasks/mask-generation/data.d.cts +1 -1
  110. package/dist/tasks/mask-generation/data.d.cts.map +1 -1
  111. package/dist/tasks/mask-generation/data.d.ts +1 -1
  112. package/dist/tasks/mask-generation/data.d.ts.map +1 -1
  113. package/dist/tasks/object-detection/data.d.cts +1 -1
  114. package/dist/tasks/object-detection/data.d.cts.map +1 -1
  115. package/dist/tasks/object-detection/data.d.ts +1 -1
  116. package/dist/tasks/object-detection/data.d.ts.map +1 -1
  117. package/dist/tasks/placeholder/data.d.cts +1 -1
  118. package/dist/tasks/placeholder/data.d.cts.map +1 -1
  119. package/dist/tasks/placeholder/data.d.ts +1 -1
  120. package/dist/tasks/placeholder/data.d.ts.map +1 -1
  121. package/dist/tasks/question-answering/data.d.cts +1 -1
  122. package/dist/tasks/question-answering/data.d.cts.map +1 -1
  123. package/dist/tasks/question-answering/data.d.ts +1 -1
  124. package/dist/tasks/question-answering/data.d.ts.map +1 -1
  125. package/dist/tasks/reinforcement-learning/data.d.cts +1 -1
  126. package/dist/tasks/reinforcement-learning/data.d.cts.map +1 -1
  127. package/dist/tasks/reinforcement-learning/data.d.ts +1 -1
  128. package/dist/tasks/reinforcement-learning/data.d.ts.map +1 -1
  129. package/dist/tasks/sentence-similarity/data.d.cts +1 -1
  130. package/dist/tasks/sentence-similarity/data.d.cts.map +1 -1
  131. package/dist/tasks/sentence-similarity/data.d.ts +1 -1
  132. package/dist/tasks/sentence-similarity/data.d.ts.map +1 -1
  133. package/dist/tasks/summarization/data.d.cts +1 -1
  134. package/dist/tasks/summarization/data.d.cts.map +1 -1
  135. package/dist/tasks/summarization/data.d.ts +1 -1
  136. package/dist/tasks/summarization/data.d.ts.map +1 -1
  137. package/dist/tasks/table-question-answering/data.d.cts +1 -1
  138. package/dist/tasks/table-question-answering/data.d.cts.map +1 -1
  139. package/dist/tasks/table-question-answering/data.d.ts +1 -1
  140. package/dist/tasks/table-question-answering/data.d.ts.map +1 -1
  141. package/dist/tasks/tabular-classification/data.d.cts +1 -1
  142. package/dist/tasks/tabular-classification/data.d.cts.map +1 -1
  143. package/dist/tasks/tabular-classification/data.d.ts +1 -1
  144. package/dist/tasks/tabular-classification/data.d.ts.map +1 -1
  145. package/dist/tasks/tabular-regression/data.d.cts +1 -1
  146. package/dist/tasks/tabular-regression/data.d.cts.map +1 -1
  147. package/dist/tasks/tabular-regression/data.d.ts +1 -1
  148. package/dist/tasks/tabular-regression/data.d.ts.map +1 -1
  149. package/dist/tasks/text-classification/data.d.cts +1 -1
  150. package/dist/tasks/text-classification/data.d.cts.map +1 -1
  151. package/dist/tasks/text-classification/data.d.ts +1 -1
  152. package/dist/tasks/text-classification/data.d.ts.map +1 -1
  153. package/dist/tasks/text-generation/data.d.cts +1 -1
  154. package/dist/tasks/text-generation/data.d.cts.map +1 -1
  155. package/dist/tasks/text-generation/data.d.ts +1 -1
  156. package/dist/tasks/text-generation/data.d.ts.map +1 -1
  157. package/dist/tasks/text-to-3d/data.d.cts +1 -1
  158. package/dist/tasks/text-to-3d/data.d.cts.map +1 -1
  159. package/dist/tasks/text-to-3d/data.d.ts +1 -1
  160. package/dist/tasks/text-to-3d/data.d.ts.map +1 -1
  161. package/dist/tasks/text-to-image/data.d.cts +1 -1
  162. package/dist/tasks/text-to-image/data.d.cts.map +1 -1
  163. package/dist/tasks/text-to-image/data.d.ts +1 -1
  164. package/dist/tasks/text-to-image/data.d.ts.map +1 -1
  165. package/dist/tasks/text-to-speech/data.d.cts +1 -1
  166. package/dist/tasks/text-to-speech/data.d.cts.map +1 -1
  167. package/dist/tasks/text-to-speech/data.d.ts +1 -1
  168. package/dist/tasks/text-to-speech/data.d.ts.map +1 -1
  169. package/dist/tasks/text-to-video/data.d.cts +1 -1
  170. package/dist/tasks/text-to-video/data.d.cts.map +1 -1
  171. package/dist/tasks/text-to-video/data.d.ts +1 -1
  172. package/dist/tasks/text-to-video/data.d.ts.map +1 -1
  173. package/dist/tasks/token-classification/data.d.cts +1 -1
  174. package/dist/tasks/token-classification/data.d.cts.map +1 -1
  175. package/dist/tasks/token-classification/data.d.ts +1 -1
  176. package/dist/tasks/token-classification/data.d.ts.map +1 -1
  177. package/dist/tasks/translation/data.d.cts +1 -1
  178. package/dist/tasks/translation/data.d.cts.map +1 -1
  179. package/dist/tasks/translation/data.d.ts +1 -1
  180. package/dist/tasks/translation/data.d.ts.map +1 -1
  181. package/dist/tasks/unconditional-image-generation/data.d.cts +1 -1
  182. package/dist/tasks/unconditional-image-generation/data.d.cts.map +1 -1
  183. package/dist/tasks/unconditional-image-generation/data.d.ts +1 -1
  184. package/dist/tasks/unconditional-image-generation/data.d.ts.map +1 -1
  185. package/dist/tasks/video-classification/data.d.cts +1 -1
  186. package/dist/tasks/video-classification/data.d.cts.map +1 -1
  187. package/dist/tasks/video-classification/data.d.ts +1 -1
  188. package/dist/tasks/video-classification/data.d.ts.map +1 -1
  189. package/dist/tasks/video-text-to-text/data.d.cts +1 -1
  190. package/dist/tasks/video-text-to-text/data.d.cts.map +1 -1
  191. package/dist/tasks/video-text-to-text/data.d.ts +1 -1
  192. package/dist/tasks/video-text-to-text/data.d.ts.map +1 -1
  193. package/dist/tasks/visual-question-answering/data.d.cts +1 -1
  194. package/dist/tasks/visual-question-answering/data.d.cts.map +1 -1
  195. package/dist/tasks/visual-question-answering/data.d.ts +1 -1
  196. package/dist/tasks/visual-question-answering/data.d.ts.map +1 -1
  197. package/dist/tasks/zero-shot-classification/data.d.cts +1 -1
  198. package/dist/tasks/zero-shot-classification/data.d.cts.map +1 -1
  199. package/dist/tasks/zero-shot-classification/data.d.ts +1 -1
  200. package/dist/tasks/zero-shot-classification/data.d.ts.map +1 -1
  201. package/dist/tasks/zero-shot-image-classification/data.d.cts +1 -1
  202. package/dist/tasks/zero-shot-image-classification/data.d.cts.map +1 -1
  203. package/dist/tasks/zero-shot-image-classification/data.d.ts +1 -1
  204. package/dist/tasks/zero-shot-image-classification/data.d.ts.map +1 -1
  205. package/dist/tasks/zero-shot-object-detection/data.d.cts +1 -1
  206. package/dist/tasks/zero-shot-object-detection/data.d.cts.map +1 -1
  207. package/dist/tasks/zero-shot-object-detection/data.d.ts +1 -1
  208. package/dist/tasks/zero-shot-object-detection/data.d.ts.map +1 -1
  209. package/dist/widget-example.d.cts +1 -1
  210. package/dist/widget-example.d.cts.map +1 -1
  211. package/dist/widget-example.d.ts +1 -1
  212. package/dist/widget-example.d.ts.map +1 -1
  213. package/package.json +1 -2
  214. package/src/default-widget-inputs.ts +2 -2
  215. package/src/index.ts +23 -19
  216. package/src/library-to-tasks.ts +2 -2
  217. package/src/local-apps.ts +3 -3
  218. package/src/model-data.ts +3 -3
  219. package/src/model-libraries-snippets.ts +3 -3
  220. package/src/model-libraries.ts +3 -3
  221. package/src/snippets/common.ts +1 -1
  222. package/src/snippets/curl.spec.ts +2 -2
  223. package/src/snippets/index.ts +5 -4
  224. package/src/snippets/inputs.ts +3 -3
  225. package/src/snippets/js.spec.ts +2 -2
  226. package/src/snippets/python.spec.ts +2 -2
  227. package/src/snippets/types.ts +1 -1
  228. package/src/tasks/audio-classification/data.ts +1 -1
  229. package/src/tasks/audio-to-audio/data.ts +1 -1
  230. package/src/tasks/automatic-speech-recognition/data.ts +1 -1
  231. package/src/tasks/depth-estimation/data.ts +1 -1
  232. package/src/tasks/document-question-answering/data.ts +1 -1
  233. package/src/tasks/feature-extraction/data.ts +1 -1
  234. package/src/tasks/fill-mask/data.ts +1 -1
  235. package/src/tasks/image-classification/data.ts +1 -1
  236. package/src/tasks/image-feature-extraction/data.ts +1 -1
  237. package/src/tasks/image-segmentation/data.ts +1 -1
  238. package/src/tasks/image-text-to-text/data.ts +1 -1
  239. package/src/tasks/image-to-3d/data.ts +1 -1
  240. package/src/tasks/image-to-image/data.ts +1 -1
  241. package/src/tasks/image-to-text/data.ts +1 -1
  242. package/src/tasks/index.ts +70 -70
  243. package/src/tasks/keypoint-detection/data.ts +1 -1
  244. package/src/tasks/mask-generation/data.ts +1 -1
  245. package/src/tasks/object-detection/data.ts +1 -1
  246. package/src/tasks/placeholder/data.ts +1 -1
  247. package/src/tasks/question-answering/data.ts +1 -1
  248. package/src/tasks/reinforcement-learning/data.ts +1 -1
  249. package/src/tasks/sentence-similarity/data.ts +1 -1
  250. package/src/tasks/summarization/data.ts +1 -1
  251. package/src/tasks/table-question-answering/data.ts +1 -1
  252. package/src/tasks/tabular-classification/data.ts +1 -1
  253. package/src/tasks/tabular-regression/data.ts +1 -1
  254. package/src/tasks/text-classification/data.ts +1 -1
  255. package/src/tasks/text-generation/data.ts +1 -1
  256. package/src/tasks/text-to-3d/data.ts +1 -1
  257. package/src/tasks/text-to-image/data.ts +1 -1
  258. package/src/tasks/text-to-speech/data.ts +1 -1
  259. package/src/tasks/text-to-video/data.ts +1 -1
  260. package/src/tasks/token-classification/data.ts +1 -1
  261. package/src/tasks/translation/data.ts +1 -1
  262. package/src/tasks/unconditional-image-generation/data.ts +1 -1
  263. package/src/tasks/video-classification/data.ts +1 -1
  264. package/src/tasks/video-text-to-text/data.ts +1 -1
  265. package/src/tasks/visual-question-answering/data.ts +1 -1
  266. package/src/tasks/zero-shot-classification/data.ts +1 -1
  267. package/src/tasks/zero-shot-image-classification/data.ts +1 -1
  268. package/src/tasks/zero-shot-object-detection/data.ts +1 -1
  269. package/src/widget-example.ts +1 -1
  270. package/tsconfig.json +2 -2
package/package.json CHANGED
@@ -1,7 +1,7 @@
1
1
  {
2
2
  "name": "@huggingface/tasks",
3
3
  "packageManager": "pnpm@8.10.5",
4
- "version": "0.13.0-test3",
4
+ "version": "0.13.0-test5",
5
5
  "description": "List of ML tasks for huggingface.co/tasks",
6
6
  "repository": "https://github.com/huggingface/huggingface.js.git",
7
7
  "publishConfig": {
@@ -12,7 +12,6 @@
12
12
  "types": "./dist/index.d.ts",
13
13
  "exports": {
14
14
  ".": {
15
- "types": "./dist/src/index.d.ts",
16
15
  "require": "./dist/index.cjs",
17
16
  "import": "./dist/index.js"
18
17
  }
@@ -1,5 +1,5 @@
1
- import type { WidgetExample } from "./widget-example";
2
- import type { WidgetType } from "./pipelines";
1
+ import type { WidgetExample } from "./widget-example.js";
2
+ import type { WidgetType } from "./pipelines.js";
3
3
 
4
4
  type LanguageCode = string;
5
5
 
package/src/index.ts CHANGED
@@ -1,7 +1,7 @@
1
- export { LIBRARY_TASK_MAPPING } from "./library-to-tasks";
2
- export { MAPPING_DEFAULT_WIDGET } from "./default-widget-inputs";
3
- export type { TaskData, TaskDemo, TaskDemoEntry, ExampleRepo } from "./tasks";
4
- export * from "./tasks";
1
+ export { LIBRARY_TASK_MAPPING } from "./library-to-tasks.js";
2
+ export { MAPPING_DEFAULT_WIDGET } from "./default-widget-inputs.js";
3
+ export type { TaskData, TaskDemo, TaskDemoEntry, ExampleRepo } from "./tasks/index.js";
4
+ export * from "./tasks/index.js";
5
5
  export {
6
6
  PIPELINE_DATA,
7
7
  PIPELINE_TYPES,
@@ -13,11 +13,15 @@ export {
13
13
  MODALITY_LABELS,
14
14
  SUBTASK_TYPES,
15
15
  PIPELINE_TYPES_SET,
16
- } from "./pipelines";
17
- export { ALL_DISPLAY_MODEL_LIBRARY_KEYS, ALL_MODEL_LIBRARY_KEYS, MODEL_LIBRARIES_UI_ELEMENTS } from "./model-libraries";
18
- export type { LibraryUiElement, ModelLibraryKey } from "./model-libraries";
19
- export type { ModelData, TransformersInfo } from "./model-data";
20
- export type { AddedToken, SpecialTokensMap, TokenizerConfig } from "./tokenizer-data";
16
+ } from "./pipelines.js";
17
+ export {
18
+ ALL_DISPLAY_MODEL_LIBRARY_KEYS,
19
+ ALL_MODEL_LIBRARY_KEYS,
20
+ MODEL_LIBRARIES_UI_ELEMENTS,
21
+ } from "./model-libraries.js";
22
+ export type { LibraryUiElement, ModelLibraryKey } from "./model-libraries.js";
23
+ export type { ModelData, TransformersInfo } from "./model-data.js";
24
+ export type { AddedToken, SpecialTokensMap, TokenizerConfig } from "./tokenizer-data.js";
21
25
  export type {
22
26
  WidgetExample,
23
27
  WidgetExampleAttribute,
@@ -38,18 +42,18 @@ export type {
38
42
  WidgetExampleOutputLabels,
39
43
  WidgetExampleOutputAnswerScore,
40
44
  WidgetExampleOutputText,
41
- } from "./widget-example";
42
- export { SPECIAL_TOKENS_ATTRIBUTES } from "./tokenizer-data";
45
+ } from "./widget-example.js";
46
+ export { SPECIAL_TOKENS_ATTRIBUTES } from "./tokenizer-data.js";
43
47
 
44
- import * as snippets from "./snippets";
45
- export * from "./gguf";
48
+ import * as snippets from "./snippets/index.js";
49
+ export * from "./gguf.js";
46
50
 
47
51
  export { snippets };
48
52
 
49
- export { SKUS, DEFAULT_MEMORY_OPTIONS } from "./hardware";
50
- export type { HardwareSpec, SkuType } from "./hardware";
51
- export { LOCAL_APPS } from "./local-apps";
52
- export type { LocalApp, LocalAppKey, LocalAppSnippet } from "./local-apps";
53
+ export { SKUS, DEFAULT_MEMORY_OPTIONS } from "./hardware.js";
54
+ export type { HardwareSpec, SkuType } from "./hardware.js";
55
+ export { LOCAL_APPS } from "./local-apps.js";
56
+ export type { LocalApp, LocalAppKey, LocalAppSnippet } from "./local-apps.js";
53
57
 
54
- export { DATASET_LIBRARIES_UI_ELEMENTS } from "./dataset-libraries";
55
- export type { DatasetLibraryUiElement, DatasetLibraryKey } from "./dataset-libraries";
58
+ export { DATASET_LIBRARIES_UI_ELEMENTS } from "./dataset-libraries.js";
59
+ export type { DatasetLibraryUiElement, DatasetLibraryKey } from "./dataset-libraries.js";
@@ -1,5 +1,5 @@
1
- import type { ModelLibraryKey } from "./model-libraries";
2
- import type { PipelineType } from "./pipelines";
1
+ import type { ModelLibraryKey } from "./model-libraries.js";
2
+ import type { PipelineType } from "./pipelines.js";
3
3
 
4
4
  /**
5
5
  * Mapping from library name to its supported tasks.
package/src/local-apps.ts CHANGED
@@ -1,6 +1,6 @@
1
- import { parseGGUFQuantLabel } from "./gguf";
2
- import type { ModelData } from "./model-data";
3
- import type { PipelineType } from "./pipelines";
1
+ import { parseGGUFQuantLabel } from "./gguf.js";
2
+ import type { ModelData } from "./model-data.js";
3
+ import type { PipelineType } from "./pipelines.js";
4
4
 
5
5
  export interface LocalAppSnippet {
6
6
  /**
package/src/model-data.ts CHANGED
@@ -1,6 +1,6 @@
1
- import type { PipelineType } from "./pipelines";
2
- import type { WidgetExample } from "./widget-example";
3
- import type { TokenizerConfig } from "./tokenizer-data";
1
+ import type { PipelineType } from "./pipelines.js";
2
+ import type { WidgetExample } from "./widget-example.js";
3
+ import type { TokenizerConfig } from "./tokenizer-data.js";
4
4
 
5
5
  /**
6
6
  * Public interface for model metadata
@@ -1,6 +1,6 @@
1
- import type { ModelData } from "./model-data";
2
- import type { WidgetExampleTextInput, WidgetExampleSentenceSimilarityInput } from "./widget-example";
3
- import { LIBRARY_TASK_MAPPING } from "./library-to-tasks";
1
+ import type { ModelData } from "./model-data.js";
2
+ import type { WidgetExampleTextInput, WidgetExampleSentenceSimilarityInput } from "./widget-example.js";
3
+ import { LIBRARY_TASK_MAPPING } from "./library-to-tasks.js";
4
4
 
5
5
  const TAG_CUSTOM_CODE = "custom_code";
6
6
 
@@ -1,6 +1,6 @@
1
- import * as snippets from "./model-libraries-snippets";
2
- import type { ModelData } from "./model-data";
3
- import type { ElasticSearchQuery } from "./model-libraries-downloads";
1
+ import * as snippets from "./model-libraries-snippets.js";
2
+ import type { ModelData } from "./model-data.js";
3
+ import type { ElasticSearchQuery } from "./model-libraries-downloads.js";
4
4
 
5
5
  /**
6
6
  * Elements configurable by a model library.
@@ -1,4 +1,4 @@
1
- import type { ChatCompletionInputMessage, GenerationParameters } from "../tasks";
1
+ import type { ChatCompletionInputMessage, GenerationParameters } from "../tasks/index.js";
2
2
 
3
3
  export function stringifyMessages(
4
4
  messages: ChatCompletionInputMessage[],
@@ -1,6 +1,6 @@
1
- import type { ModelDataMinimal } from "./types";
1
+ import type { ModelDataMinimal } from "./types.js";
2
2
  import { describe, expect, it } from "vitest";
3
- import { snippetTextGeneration } from "./curl";
3
+ import { snippetTextGeneration } from "./curl.js";
4
4
 
5
5
  describe("inference API snippets", () => {
6
6
  it("conversational llm", async () => {
@@ -1,6 +1,7 @@
1
- import * as inputs from "./inputs";
2
- import * as curl from "./curl";
3
- import * as python from "./python";
4
- import * as js from "./js";
1
+ import * as inputs from "./inputs.js";
2
+ import * as curl from "./curl.js";
3
+ import * as python from "./python.js";
4
+ import * as js from "./js.js";
5
+ export * from "./types.js";
5
6
 
6
7
  export { inputs, curl, python, js };
@@ -1,6 +1,6 @@
1
- import type { PipelineType } from "../pipelines";
2
- import type { ChatCompletionInputMessage } from "../tasks";
3
- import type { ModelDataMinimal } from "./types";
1
+ import type { PipelineType } from "../pipelines.js";
2
+ import type { ChatCompletionInputMessage } from "../tasks/index.js";
3
+ import type { ModelDataMinimal } from "./types.js";
4
4
 
5
5
  const inputsZeroShotClassification = () =>
6
6
  `"Hi, I recently bought a device from your company but it is not working as advertised and I would like to get reimbursed!"`;
@@ -1,6 +1,6 @@
1
- import type { InferenceSnippet, ModelDataMinimal } from "./types";
1
+ import type { InferenceSnippet, ModelDataMinimal } from "./types.js";
2
2
  import { describe, expect, it } from "vitest";
3
- import { snippetTextGeneration } from "./js";
3
+ import { snippetTextGeneration } from "./js.js";
4
4
 
5
5
  describe("inference API snippets", () => {
6
6
  it("conversational llm", async () => {
@@ -1,6 +1,6 @@
1
- import type { ModelDataMinimal } from "./types";
1
+ import type { ModelDataMinimal } from "./types.js";
2
2
  import { describe, expect, it } from "vitest";
3
- import { snippetConversational } from "./python";
3
+ import { snippetConversational } from "./python.js";
4
4
 
5
5
  describe("inference API snippets", () => {
6
6
  it("conversational llm", async () => {
@@ -1,4 +1,4 @@
1
- import type { ModelData } from "../model-data";
1
+ import type { ModelData } from "../model-data.js";
2
2
 
3
3
  /**
4
4
  * Minimal model data required for snippets.
@@ -1,4 +1,4 @@
1
- import type { TaskDataCustom } from "..";
1
+ import type { TaskDataCustom } from "../index.js";
2
2
 
3
3
  const taskData: TaskDataCustom = {
4
4
  datasets: [
@@ -1,4 +1,4 @@
1
- import type { TaskDataCustom } from "..";
1
+ import type { TaskDataCustom } from "../index.js";
2
2
 
3
3
  const taskData: TaskDataCustom = {
4
4
  datasets: [
@@ -1,4 +1,4 @@
1
- import type { TaskDataCustom } from "..";
1
+ import type { TaskDataCustom } from "../index.js";
2
2
 
3
3
  const taskData: TaskDataCustom = {
4
4
  datasets: [
@@ -1,4 +1,4 @@
1
- import type { TaskDataCustom } from "..";
1
+ import type { TaskDataCustom } from "../index.js";
2
2
 
3
3
  const taskData: TaskDataCustom = {
4
4
  datasets: [
@@ -1,4 +1,4 @@
1
- import type { TaskDataCustom } from "..";
1
+ import type { TaskDataCustom } from "../index.js";
2
2
 
3
3
  const taskData: TaskDataCustom = {
4
4
  datasets: [
@@ -1,4 +1,4 @@
1
- import type { TaskDataCustom } from "..";
1
+ import type { TaskDataCustom } from "../index.js";
2
2
 
3
3
  const taskData: TaskDataCustom = {
4
4
  datasets: [
@@ -1,4 +1,4 @@
1
- import type { TaskDataCustom } from "..";
1
+ import type { TaskDataCustom } from "../index.js";
2
2
 
3
3
  const taskData: TaskDataCustom = {
4
4
  datasets: [
@@ -1,4 +1,4 @@
1
- import type { TaskDataCustom } from "..";
1
+ import type { TaskDataCustom } from "../index.js";
2
2
 
3
3
  const taskData: TaskDataCustom = {
4
4
  datasets: [
@@ -1,4 +1,4 @@
1
- import type { TaskDataCustom } from "..";
1
+ import type { TaskDataCustom } from "../index.js";
2
2
 
3
3
  const taskData: TaskDataCustom = {
4
4
  datasets: [
@@ -1,4 +1,4 @@
1
- import type { TaskDataCustom } from "..";
1
+ import type { TaskDataCustom } from "../index.js";
2
2
 
3
3
  const taskData: TaskDataCustom = {
4
4
  datasets: [
@@ -1,4 +1,4 @@
1
- import type { TaskDataCustom } from "..";
1
+ import type { TaskDataCustom } from "../index.js";
2
2
 
3
3
  const taskData: TaskDataCustom = {
4
4
  datasets: [
@@ -1,4 +1,4 @@
1
- import type { TaskDataCustom } from "..";
1
+ import type { TaskDataCustom } from "../index.js";
2
2
 
3
3
  const taskData: TaskDataCustom = {
4
4
  datasets: [
@@ -1,4 +1,4 @@
1
- import type { TaskDataCustom } from "..";
1
+ import type { TaskDataCustom } from "../index.js";
2
2
 
3
3
  const taskData: TaskDataCustom = {
4
4
  datasets: [
@@ -1,4 +1,4 @@
1
- import type { TaskDataCustom } from "..";
1
+ import type { TaskDataCustom } from "../index.js";
2
2
 
3
3
  const taskData: TaskDataCustom = {
4
4
  datasets: [
@@ -1,49 +1,49 @@
1
- import type { PipelineType } from "../pipelines";
2
- import { PIPELINE_DATA } from "../pipelines";
1
+ import type { PipelineType } from "../pipelines.js";
2
+ import { PIPELINE_DATA } from "../pipelines.js";
3
3
 
4
- import audioClassification from "./audio-classification/data";
5
- import audioToAudio from "./audio-to-audio/data";
6
- import automaticSpeechRecognition from "./automatic-speech-recognition/data";
7
- import documentQuestionAnswering from "./document-question-answering/data";
8
- import featureExtraction from "./feature-extraction/data";
9
- import fillMask from "./fill-mask/data";
10
- import imageClassification from "./image-classification/data";
11
- import imageFeatureExtraction from "./image-feature-extraction/data";
12
- import imageToImage from "./image-to-image/data";
13
- import imageToText from "./image-to-text/data";
14
- import imageTextToText from "./image-text-to-text/data";
15
- import imageSegmentation from "./image-segmentation/data";
16
- import maskGeneration from "./mask-generation/data";
17
- import objectDetection from "./object-detection/data";
18
- import depthEstimation from "./depth-estimation/data";
19
- import placeholder from "./placeholder/data";
20
- import reinforcementLearning from "./reinforcement-learning/data";
21
- import questionAnswering from "./question-answering/data";
22
- import sentenceSimilarity from "./sentence-similarity/data";
23
- import summarization from "./summarization/data";
24
- import tableQuestionAnswering from "./table-question-answering/data";
25
- import tabularClassification from "./tabular-classification/data";
26
- import tabularRegression from "./tabular-regression/data";
27
- import textToImage from "./text-to-image/data";
28
- import textToSpeech from "./text-to-speech/data";
29
- import tokenClassification from "./token-classification/data";
30
- import translation from "./translation/data";
31
- import textClassification from "./text-classification/data";
32
- import textGeneration from "./text-generation/data";
33
- import textToVideo from "./text-to-video/data";
34
- import unconditionalImageGeneration from "./unconditional-image-generation/data";
35
- import videoClassification from "./video-classification/data";
36
- import visualQuestionAnswering from "./visual-question-answering/data";
37
- import zeroShotClassification from "./zero-shot-classification/data";
38
- import zeroShotImageClassification from "./zero-shot-image-classification/data";
39
- import zeroShotObjectDetection from "./zero-shot-object-detection/data";
40
- import imageTo3D from "./image-to-3d/data";
41
- import textTo3D from "./text-to-3d/data";
42
- import keypointDetection from "./keypoint-detection/data";
43
- import videoTextToText from "./video-text-to-text/data";
4
+ import audioClassification from "./audio-classification/data.js";
5
+ import audioToAudio from "./audio-to-audio/data.js";
6
+ import automaticSpeechRecognition from "./automatic-speech-recognition/data.js";
7
+ import documentQuestionAnswering from "./document-question-answering/data.js";
8
+ import featureExtraction from "./feature-extraction/data.js";
9
+ import fillMask from "./fill-mask/data.js";
10
+ import imageClassification from "./image-classification/data.js";
11
+ import imageFeatureExtraction from "./image-feature-extraction/data.js";
12
+ import imageToImage from "./image-to-image/data.js";
13
+ import imageToText from "./image-to-text/data.js";
14
+ import imageTextToText from "./image-text-to-text/data.js";
15
+ import imageSegmentation from "./image-segmentation/data.js";
16
+ import maskGeneration from "./mask-generation/data.js";
17
+ import objectDetection from "./object-detection/data.js";
18
+ import depthEstimation from "./depth-estimation/data.js";
19
+ import placeholder from "./placeholder/data.js";
20
+ import reinforcementLearning from "./reinforcement-learning/data.js";
21
+ import questionAnswering from "./question-answering/data.js";
22
+ import sentenceSimilarity from "./sentence-similarity/data.js";
23
+ import summarization from "./summarization/data.js";
24
+ import tableQuestionAnswering from "./table-question-answering/data.js";
25
+ import tabularClassification from "./tabular-classification/data.js";
26
+ import tabularRegression from "./tabular-regression/data.js";
27
+ import textToImage from "./text-to-image/data.js";
28
+ import textToSpeech from "./text-to-speech/data.js";
29
+ import tokenClassification from "./token-classification/data.js";
30
+ import translation from "./translation/data.js";
31
+ import textClassification from "./text-classification/data.js";
32
+ import textGeneration from "./text-generation/data.js";
33
+ import textToVideo from "./text-to-video/data.js";
34
+ import unconditionalImageGeneration from "./unconditional-image-generation/data.js";
35
+ import videoClassification from "./video-classification/data.js";
36
+ import visualQuestionAnswering from "./visual-question-answering/data.js";
37
+ import zeroShotClassification from "./zero-shot-classification/data.js";
38
+ import zeroShotImageClassification from "./zero-shot-image-classification/data.js";
39
+ import zeroShotObjectDetection from "./zero-shot-object-detection/data.js";
40
+ import imageTo3D from "./image-to-3d/data.js";
41
+ import textTo3D from "./text-to-3d/data.js";
42
+ import keypointDetection from "./keypoint-detection/data.js";
43
+ import videoTextToText from "./video-text-to-text/data.js";
44
44
 
45
- export type * from "./audio-classification/inference";
46
- export type * from "./automatic-speech-recognition/inference";
45
+ export type * from "./audio-classification/inference.js";
46
+ export type * from "./automatic-speech-recognition/inference.js";
47
47
  export type {
48
48
  ChatCompletionInput,
49
49
  ChatCompletionInputMessage,
@@ -53,36 +53,36 @@ export type {
53
53
  ChatCompletionStreamOutput,
54
54
  ChatCompletionStreamOutputChoice,
55
55
  ChatCompletionStreamOutputDelta,
56
- } from "./chat-completion/inference";
57
- export type * from "./document-question-answering/inference";
58
- export type * from "./feature-extraction/inference";
59
- export type * from "./fill-mask/inference";
56
+ } from "./chat-completion/inference.js";
57
+ export type * from "./document-question-answering/inference.js";
58
+ export type * from "./feature-extraction/inference.js";
59
+ export type * from "./fill-mask/inference.js";
60
60
  export type {
61
61
  ImageClassificationInput,
62
62
  ImageClassificationOutput,
63
63
  ImageClassificationOutputElement,
64
64
  ImageClassificationParameters,
65
- } from "./image-classification/inference";
66
- export type * from "./image-to-image/inference";
67
- export type { ImageToTextInput, ImageToTextOutput, ImageToTextParameters } from "./image-to-text/inference";
68
- export type * from "./image-segmentation/inference";
69
- export type * from "./object-detection/inference";
70
- export type * from "./depth-estimation/inference";
71
- export type * from "./question-answering/inference";
72
- export type * from "./sentence-similarity/inference";
73
- export type * from "./summarization/inference";
74
- export type * from "./table-question-answering/inference";
75
- export type { TextToImageInput, TextToImageOutput, TextToImageParameters } from "./text-to-image/inference";
76
- export type { TextToSpeechParameters, TextToSpeechInput, TextToSpeechOutput } from "./text-to-speech/inference";
77
- export type * from "./token-classification/inference";
78
- export type { TranslationInput, TranslationOutput } from "./translation/inference";
65
+ } from "./image-classification/inference.js";
66
+ export type * from "./image-to-image/inference.js";
67
+ export type { ImageToTextInput, ImageToTextOutput, ImageToTextParameters } from "./image-to-text/inference.js";
68
+ export type * from "./image-segmentation/inference.js";
69
+ export type * from "./object-detection/inference.js";
70
+ export type * from "./depth-estimation/inference.js";
71
+ export type * from "./question-answering/inference.js";
72
+ export type * from "./sentence-similarity/inference.js";
73
+ export type * from "./summarization/inference.js";
74
+ export type * from "./table-question-answering/inference.js";
75
+ export type { TextToImageInput, TextToImageOutput, TextToImageParameters } from "./text-to-image/inference.js";
76
+ export type { TextToSpeechParameters, TextToSpeechInput, TextToSpeechOutput } from "./text-to-speech/inference.js";
77
+ export type * from "./token-classification/inference.js";
78
+ export type { TranslationInput, TranslationOutput } from "./translation/inference.js";
79
79
  export type {
80
80
  ClassificationOutputTransform,
81
81
  TextClassificationInput,
82
82
  TextClassificationOutput,
83
83
  TextClassificationOutputElement,
84
84
  TextClassificationParameters,
85
- } from "./text-classification/inference";
85
+ } from "./text-classification/inference.js";
86
86
  export type {
87
87
  TextGenerationOutputFinishReason,
88
88
  TextGenerationOutputPrefillToken,
@@ -94,20 +94,20 @@ export type {
94
94
  TextGenerationOutputToken,
95
95
  TextGenerationStreamOutputStreamDetails,
96
96
  TextGenerationStreamOutput,
97
- } from "./text-generation/inference";
98
- export type * from "./video-classification/inference";
99
- export type * from "./visual-question-answering/inference";
100
- export type * from "./zero-shot-classification/inference";
101
- export type * from "./zero-shot-image-classification/inference";
97
+ } from "./text-generation/inference.js";
98
+ export type * from "./video-classification/inference.js";
99
+ export type * from "./visual-question-answering/inference.js";
100
+ export type * from "./zero-shot-classification/inference.js";
101
+ export type * from "./zero-shot-image-classification/inference.js";
102
102
  export type {
103
103
  BoundingBox,
104
104
  ZeroShotObjectDetectionInput,
105
105
  ZeroShotObjectDetectionInputData,
106
106
  ZeroShotObjectDetectionOutput,
107
107
  ZeroShotObjectDetectionOutputElement,
108
- } from "./zero-shot-object-detection/inference";
108
+ } from "./zero-shot-object-detection/inference.js";
109
109
 
110
- import type { ModelLibraryKey } from "../model-libraries";
110
+ import type { ModelLibraryKey } from "../model-libraries.js";
111
111
 
112
112
  /**
113
113
  * Model libraries compatible with each ML task
@@ -1,4 +1,4 @@
1
- import type { TaskDataCustom } from "..";
1
+ import type { TaskDataCustom } from "../index.js";
2
2
 
3
3
  const taskData: TaskDataCustom = {
4
4
  datasets: [
@@ -1,4 +1,4 @@
1
- import type { TaskDataCustom } from "..";
1
+ import type { TaskDataCustom } from "../index.js";
2
2
 
3
3
  const taskData: TaskDataCustom = {
4
4
  datasets: [],
@@ -1,4 +1,4 @@
1
- import type { TaskDataCustom } from "..";
1
+ import type { TaskDataCustom } from "../index.js";
2
2
 
3
3
  const taskData: TaskDataCustom = {
4
4
  datasets: [
@@ -1,4 +1,4 @@
1
- import type { TaskDataCustom } from "..";
1
+ import type { TaskDataCustom } from "../index.js";
2
2
 
3
3
  const taskData: TaskDataCustom = {
4
4
  datasets: [],
@@ -1,4 +1,4 @@
1
- import type { TaskDataCustom } from "..";
1
+ import type { TaskDataCustom } from "../index.js";
2
2
 
3
3
  const taskData: TaskDataCustom = {
4
4
  datasets: [
@@ -1,4 +1,4 @@
1
- import type { TaskDataCustom } from "..";
1
+ import type { TaskDataCustom } from "../index.js";
2
2
 
3
3
  const taskData: TaskDataCustom = {
4
4
  datasets: [
@@ -1,4 +1,4 @@
1
- import type { TaskDataCustom } from "..";
1
+ import type { TaskDataCustom } from "../index.js";
2
2
 
3
3
  const taskData: TaskDataCustom = {
4
4
  datasets: [
@@ -1,4 +1,4 @@
1
- import type { TaskDataCustom } from "..";
1
+ import type { TaskDataCustom } from "../index.js";
2
2
 
3
3
  const taskData: TaskDataCustom = {
4
4
  canonicalId: "text2text-generation",
@@ -1,4 +1,4 @@
1
- import type { TaskDataCustom } from "..";
1
+ import type { TaskDataCustom } from "../index.js";
2
2
 
3
3
  const taskData: TaskDataCustom = {
4
4
  datasets: [
@@ -1,4 +1,4 @@
1
- import type { TaskDataCustom } from "..";
1
+ import type { TaskDataCustom } from "../index.js";
2
2
 
3
3
  const taskData: TaskDataCustom = {
4
4
  datasets: [
@@ -1,4 +1,4 @@
1
- import type { TaskDataCustom } from "..";
1
+ import type { TaskDataCustom } from "../index.js";
2
2
 
3
3
  const taskData: TaskDataCustom = {
4
4
  datasets: [
@@ -1,4 +1,4 @@
1
- import type { TaskDataCustom } from "..";
1
+ import type { TaskDataCustom } from "../index.js";
2
2
 
3
3
  const taskData: TaskDataCustom = {
4
4
  datasets: [
@@ -1,4 +1,4 @@
1
- import type { TaskDataCustom } from "..";
1
+ import type { TaskDataCustom } from "../index.js";
2
2
 
3
3
  const taskData: TaskDataCustom = {
4
4
  datasets: [
@@ -1,4 +1,4 @@
1
- import type { TaskDataCustom } from "..";
1
+ import type { TaskDataCustom } from "../index.js";
2
2
 
3
3
  const taskData: TaskDataCustom = {
4
4
  datasets: [
@@ -1,4 +1,4 @@
1
- import type { TaskDataCustom } from "..";
1
+ import type { TaskDataCustom } from "../index.js";
2
2
 
3
3
  const taskData: TaskDataCustom = {
4
4
  datasets: [
@@ -1,4 +1,4 @@
1
- import type { TaskDataCustom } from "..";
1
+ import type { TaskDataCustom } from "../index.js";
2
2
 
3
3
  const taskData: TaskDataCustom = {
4
4
  canonicalId: "text-to-audio",
@@ -1,4 +1,4 @@
1
- import type { TaskDataCustom } from "..";
1
+ import type { TaskDataCustom } from "../index.js";
2
2
 
3
3
  const taskData: TaskDataCustom = {
4
4
  datasets: [
@@ -1,4 +1,4 @@
1
- import type { TaskDataCustom } from "..";
1
+ import type { TaskDataCustom } from "../index.js";
2
2
 
3
3
  const taskData: TaskDataCustom = {
4
4
  datasets: [