model-metadata/code_python_files

1# /// script2# requires-python = ">=3.12"3# dependencies = [4#     "torch",5#     "torchvision",6#     "transformers",7#     "diffusers",8#     "sentence-transformers",9#     "accelerate",10#     "peft",11#     "slack-sdk",12# ]13# ///14 15try:16    # Use a pipeline as a high-level helper17    from transformers import pipeline18    from transformers import AutoTokenizer19    20    model_id = "HuggingFaceTB/SmolLM3-3B"21    22    tokenizer = AutoTokenizer.from_pretrained(model_id)23    24    pipe = pipeline("text-generation", model=model_id, tokenizer=tokenizer)25    with open('HuggingFaceTB_SmolLM3-3B_0.txt', 'w', encoding='utf-8') as f:26        f.write('Everything was good in HuggingFaceTB_SmolLM3-3B_0.txt')27except Exception as e:28    import os29    from slack_sdk import WebClient30    client = WebClient(token=os.environ['SLACK_TOKEN'])31    client.chat_postMessage(32        channel='#hub-model-metadata-snippets-sprint',33        text='Problem in <https://huggingface.co/datasets/model-metadata/code_execution_files/blob/main/HuggingFaceTB_SmolLM3-3B_0.txt|HuggingFaceTB_SmolLM3-3B_0.txt>',34    )35 36    with open('HuggingFaceTB_SmolLM3-3B_0.txt', 'a', encoding='utf-8') as f:37        import traceback38        f.write('''```CODE: 39# Use a pipeline as a high-level helper40from transformers import pipeline41from transformers import AutoTokenizer42 43model_id = "HuggingFaceTB/SmolLM3-3B"44 45tokenizer = AutoTokenizer.from_pretrained(model_id)46 47pipe = pipeline("text-generation", model=model_id, tokenizer=tokenizer)48```49 50ERROR: 51''')52        traceback.print_exc(file=f)53    54finally:55    from huggingface_hub import upload_file56    upload_file(57        path_or_fileobj='HuggingFaceTB_SmolLM3-3B_0.txt',58        repo_id='model-metadata/code_execution_files',59        path_in_repo='HuggingFaceTB_SmolLM3-3B_0.txt',60        repo_type='dataset',61    )62