model-metadata/custom_code_py_files

1# /// script2# requires-python = ">=3.12"3# dependencies = [4#     "transformers",5#     "torch",6# ]7# ///8 9try:10    # Load model directly11    from transformers import AutoModel12    model = AutoModel.from_pretrained("Kwai-Keye/Keye-VL-8B-Preview", trust_remote_code=True)13    with open('Keye-VL-8B-Preview_0.txt', 'w') as f:14        f.write('Everything was good in Keye-VL-8B-Preview_0')15except Exception as e:16    with open('Keye-VL-8B-Preview_0.txt', 'w') as f:17        import traceback18        traceback.print_exc(file=f)19finally:20    from huggingface_hub import upload_file21    upload_file(22        path_or_fileobj='Keye-VL-8B-Preview_0.txt',23        repo_id='model-metadata/custom_code_execution_files',24        path_in_repo='Keye-VL-8B-Preview_0.txt',25        repo_type='dataset',26    )