model-metadata/code_python_files

1# /// script2# requires-python = ">=3.12"3# dependencies = [4#     "torch",5#     "torchvision",6#     "transformers",7#     "accelerate",8#     "peft",9#     "slack-sdk",10# ]11# ///12 13try:14    # Load model directly15    from transformers import AutoProcessor, AutoModelForVision2Seq16    17    processor = AutoProcessor.from_pretrained("datalab-to/chandra")18    model = AutoModelForVision2Seq.from_pretrained("datalab-to/chandra")19    with open('datalab-to_chandra_2.txt', 'w', encoding='utf-8') as f:20        f.write('Everything was good in datalab-to_chandra_2.txt')21except Exception as e:22    import os23    from slack_sdk import WebClient24    client = WebClient(token=os.environ['SLACK_TOKEN'])25    client.chat_postMessage(26        channel='#exp-slack-alerts',27        text='Problem in <https://huggingface.co/datasets/model-metadata/code_execution_files/blob/main/datalab-to_chandra_2.txt|datalab-to_chandra_2.txt>',28    )29 30    with open('datalab-to_chandra_2.txt', 'a', encoding='utf-8') as f:31        import traceback32        f.write('''```CODE: 33# Load model directly34from transformers import AutoProcessor, AutoModelForVision2Seq35 36processor = AutoProcessor.from_pretrained("datalab-to/chandra")37model = AutoModelForVision2Seq.from_pretrained("datalab-to/chandra")38```39 40ERROR: 41''')42        traceback.print_exc(file=f)43    44finally:45    from huggingface_hub import upload_file46    upload_file(47        path_or_fileobj='datalab-to_chandra_2.txt',48        repo_id='model-metadata/code_execution_files',49        path_in_repo='datalab-to_chandra_2.txt',50        repo_type='dataset',51    )52