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 gliner2 import GLiNER224    25    model = GLiNER2.from_pretrained("fastino/gliner2-base-v1")26    27    # Extract entities28    text = "Apple CEO Tim Cook announced iPhone 15 in Cupertino yesterday."29    result = extractor.extract_entities(text, ["company", "person", "product", "location"])30    31    print(result)32    with open('fastino_gliner2-base-v1_0.txt', 'w', encoding='utf-8') as f:33        f.write('Everything was good in fastino_gliner2-base-v1_0.txt')34except Exception as e:35    import os36    from slack_sdk import WebClient37    client = WebClient(token=os.environ['SLACK_TOKEN'])38    client.chat_postMessage(39        channel='#hub-model-metadata-snippets-sprint',40        text='Problem in <https://huggingface.co/datasets/model-metadata/code_execution_files/blob/main/fastino_gliner2-base-v1_0.txt|fastino_gliner2-base-v1_0.txt>',41    )42 43    with open('fastino_gliner2-base-v1_0.txt', 'a', encoding='utf-8') as f:44        import traceback45        f.write('''```CODE: 46from gliner2 import GLiNER247 48model = GLiNER2.from_pretrained("fastino/gliner2-base-v1")49 50# Extract entities51text = "Apple CEO Tim Cook announced iPhone 15 in Cupertino yesterday."52result = extractor.extract_entities(text, ["company", "person", "product", "location"])53 54print(result)55```56 57ERROR: 58''')59        traceback.print_exc(file=f)60    61finally:62    from huggingface_hub import upload_file63    upload_file(64        path_or_fileobj='fastino_gliner2-base-v1_0.txt',65        repo_id='model-metadata/code_execution_files',66        path_in_repo='fastino_gliner2-base-v1_0.txt',67        repo_type='dataset',68    )69