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";
