model-metadata/code_python_files

1# /// script2# requires-python = ">=3.12"3# dependencies = [4#     "numpy",5#     "einops",6#     "pandas",7#     "protobuf",8#     "torch",9#     "torchvision",10#     "transformers",11#     "timm",12#     "diffusers",13#     "sentence-transformers",14#     "accelerate",15#     "peft",16#     "slack-sdk",17# ]18# ///19 20try:21    from diffusers import DiffusionPipeline22    from diffusers.utils import load_image23    24    pipe = DiffusionPipeline.from_pretrained("dx8152/White_to_Scene")25    26    prompt = "Turn this cat into a dog"27    input_image = load_image("https://huggingface.co/datasets/huggingface/documentation-images/resolve/main/diffusers/cat.png")28    29    image = pipe(image=input_image, prompt=prompt).images[0]30    with open('dx8152_White_to_Scene_0.txt', 'w', encoding='utf-8') as f:31        f.write('Everything was good in dx8152_White_to_Scene_0.txt')32except Exception as e:33    import os34    from slack_sdk import WebClient35    client = WebClient(token=os.environ['SLACK_TOKEN'])36    client.chat_postMessage(37        channel='#hub-model-metadata-snippets-sprint',38        text='Problem in <https://huggingface.co/datasets/model-metadata/code_execution_files/blob/main/dx8152_White_to_Scene_0.txt|dx8152_White_to_Scene_0.txt>',39    )40 41    with open('dx8152_White_to_Scene_0.txt', 'a', encoding='utf-8') as f:42        import traceback43        f.write('''```CODE: 44from diffusers import DiffusionPipeline45from diffusers.utils import load_image46 47pipe = DiffusionPipeline.from_pretrained("dx8152/White_to_Scene")48 49prompt = "Turn this cat into a dog"50input_image = load_image("https://huggingface.co/datasets/huggingface/documentation-images/resolve/main/diffusers/cat.png")51 52image = pipe(image=input_image, prompt=prompt).images[0]53```54 55ERROR: 56''')57        traceback.print_exc(file=f)58    59finally:60    from huggingface_hub import upload_file61    upload_file(62        path_or_fileobj='dx8152_White_to_Scene_0.txt',63        repo_id='model-metadata/code_execution_files',64        path_in_repo='dx8152_White_to_Scene_0.txt',65        repo_type='dataset',66    )67