54 lines
1.2 KiB
TypeScript
54 lines
1.2 KiB
TypeScript
export type SearchModelParamHf = {
|
|
search?: {
|
|
owner?: string
|
|
task?: Task
|
|
}
|
|
credentials?: {
|
|
accessToken: string
|
|
}
|
|
limit: number
|
|
}
|
|
|
|
export type Task =
|
|
| 'text-classification'
|
|
| 'token-classification'
|
|
| 'table-question-answering'
|
|
| 'question-answering'
|
|
| 'zero-shot-classification'
|
|
| 'translation'
|
|
| 'summarization'
|
|
| 'conversational'
|
|
| 'feature-extraction'
|
|
| 'text-generation'
|
|
| 'text2text-generation'
|
|
| 'fill-mask'
|
|
| 'sentence-similarity'
|
|
| 'text-to-speech'
|
|
| 'automatic-speech-recognition'
|
|
| 'audio-to-audio'
|
|
| 'audio-classification'
|
|
| 'voice-activity-detection'
|
|
| 'depth-estimation'
|
|
| 'image-classification'
|
|
| 'object-detection'
|
|
| 'image-segmentation'
|
|
| 'text-to-image'
|
|
| 'image-to-text'
|
|
| 'image-to-image'
|
|
| 'unconditional-image-generation'
|
|
| 'video-classification'
|
|
| 'reinforcement-learning'
|
|
| 'robotics'
|
|
| 'tabular-classification'
|
|
| 'tabular-regression'
|
|
| 'tabular-to-text'
|
|
| 'table-to-text'
|
|
| 'multiple-choice'
|
|
| 'text-retrieval'
|
|
| 'time-series-forecasting'
|
|
| 'visual-question-answering'
|
|
| 'document-question-answering'
|
|
| 'zero-shot-image-classification'
|
|
| 'graph-ml'
|
|
| 'other'
|