model-metadata/code_python_files

1# /// script2# requires-python = ">=3.12"3# dependencies = [4#     "numpy",5#     "einops",6#     "pandas",7#     "matplotlib",8#     "protobuf",9#     "torch",10#     "sentencepiece",11#     "torchvision",12#     "transformers",13#     "timm",14#     "diffusers",15#     "sentence-transformers",16#     "accelerate",17#     "peft",18#     "slack-sdk",19# ]20# ///21 22try:23    from sentence_transformers import SentenceTransformer24    25    model = SentenceTransformer("MongoDB/mdbr-leaf-ir")26    27    sentences = [28        "That is a happy person",29        "That is a happy dog",30        "That is a very happy person",31        "Today is a sunny day"32    ]33    embeddings = model.encode(sentences)34    35    similarities = model.similarity(embeddings, embeddings)36    print(similarities.shape)37    # [4, 4]38    with open('MongoDB_mdbr-leaf-ir_0.txt', 'w', encoding='utf-8') as f:39        f.write('Everything was good in MongoDB_mdbr-leaf-ir_0.txt')40except Exception as e:41    import os42    from slack_sdk import WebClient43    client = WebClient(token=os.environ['SLACK_TOKEN'])44    client.chat_postMessage(45        channel='#hub-model-metadata-snippets-sprint',46        text='Problem in <https://huggingface.co/datasets/model-metadata/code_execution_files/blob/main/MongoDB_mdbr-leaf-ir_0.txt|MongoDB_mdbr-leaf-ir_0.txt>',47    )48 49    with open('MongoDB_mdbr-leaf-ir_0.txt', 'a', encoding='utf-8') as f:50        import traceback51        f.write('''```CODE: 52from sentence_transformers import SentenceTransformer53 54model = SentenceTransformer("MongoDB/mdbr-leaf-ir")55 56sentences = [57    "That is a happy person",58    "That is a happy dog",59    "That is a very happy person",60    "Today is a sunny day"61]62embeddings = model.encode(sentences)63 64similarities = model.similarity(embeddings, embeddings)65print(similarities.shape)66# [4, 4]67```68 69ERROR: 70''')71        traceback.print_exc(file=f)72    73finally:74    from huggingface_hub import upload_file75    upload_file(76        path_or_fileobj='MongoDB_mdbr-leaf-ir_0.txt',77        repo_id='model-metadata/code_execution_files',78        path_in_repo='MongoDB_mdbr-leaf-ir_0.txt',79        repo_type='dataset',80    )81