export const Task = Object.freeze({
    TEXT_CLASSIFICATION: "text-classification",
    TOKEN_CLASSIFICATION: "token-classification",
    TABLE_QUESTION_ANSWERING: "table-question-answering",
    QUESTION_ANSWERING: "question-answering",
    ZERO_SHOT_CLASSIFICATION: "zero-shot-classification",
    TRANSLATION: "translation",
    SUMMARIZATION: "summarization",
    CONVERSATIONAL: "conversational",
    FEATURE_EXTRACTION: "feature-extraction",
    TEXT_GENERATION: "text-generation",
    TEXT2TEXT_GENERATION: "text2text-generation",
    FILL_MASK: "fill-mask",
    SENTENCE_SIMILARITY: "sentence-similarity",
    TEXT_TO_SPEECH: "text-to-speech",
    AUTOMATIC_SPEECH_RECOGNITION: "automatic-speech-recognition",
    AUDIO_TO_AUDIO: "audio-to-audio",
    AUDIO_CLASSIFICATION: "audio-classification",
    VOICE_ACTIVITY_DETECTION: "voice-activity-detection",
    DEPTH_ESTIMATION: "depth-estimation",
    IMAGE_CLASSIFICATION: "image-classification",
    OBJECT_DETECTION: "object-detection",
    IMAGE_SEGMENTATION: "image-segmentation",
    TEXT_TO_IMAGE: "text-to-image",
    IMAGE_TO_TEXT: "image-to-text",
    IMAGE_TO_IMAGE: "image-to-image",
    UNCONDITIONAL_IMAGE_GENERATION: "unconditional-image-generation",
    VIDEO_CLASSIFICATION: "video-classification",
    REINFORCEMENT_LEARNING: "reinforcement-learning",
    ROBOTICS: "robotics",
    TABULAR_CLASSIFICATION: "tabular-classification",
    TABULAR_REGRESSION: "tabular-regression",
    TABULAR_TO_TEXT: "tabular-to-text",
    TABLE_TO_TEXT: "table-to-text",
    MULTIPLE_CHOICE: "multiple-choice",
    TEXT_RETRIEVAL: "text-retrieval",
    TIME_SERIES_FORECASTING: "time-series-forecasting",
    VISUAL_QUESTION_ANSWERING: "visual-question-answering",
    DOCUMENT_QUESTION_ANSWERING: "document-question-answering",
    ZERO_SHOT_IMAGE_CLASSIFICATION: "zero-shot-image-classification",
    GRAPH_ML: "graph-ml",
    OTHER: "other",
});

export const textGenerationModels = [
    "bigcode/starcoder",
    "meta-llama/Llama-2-7b-chat-hf",
    "meta-llama/Llama-2-70b-chat-hf",
    "tiiuae/falcon-7b",
    "meta-llama/Llama-2-7b-hf",
    "tiiuae/falcon-7b-instruct",
    "anon8231489123/vicuna-13b-GPTQ-4bit-128g",
    "meta-llama/Llama-2-13b-chat-hf",
    "meta-llama/Llama-2-70b-hf",
    "Gustavosta/MagicPrompt-Stable-Diffusion"
]