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    from diffusers import DiffusionPipeline15    from diffusers.utils import load_image16    17    pipe = DiffusionPipeline.from_pretrained("Qwen/Qwen-Image-Edit-2509")18    pipe.load_lora_weights("dx8152/Relight")19    20    prompt = "Turn this cat into a dog"21    input_image = load_image("https://huggingface.co/datasets/huggingface/documentation-images/resolve/main/diffusers/cat.png")22    23    image = pipe(image=input_image, prompt=prompt).images[0]24    with open('dx8152_Relight_2.txt', 'w', encoding='utf-8') as f:25        f.write('Everything was good in dx8152_Relight_2.txt')26except Exception as e:27    import os28    from slack_sdk import WebClient29    client = WebClient(token=os.environ['SLACK_TOKEN'])30    client.chat_postMessage(31        channel='#exp-slack-alerts',32        text='Problem in <https://huggingface.co/datasets/model-metadata/code_execution_files/blob/main/dx8152_Relight_2.txt|dx8152_Relight_2.txt>',33    )34 35    with open('dx8152_Relight_2.txt', 'a', encoding='utf-8') as f:36        import traceback37        f.write('''```CODE: 38from diffusers import DiffusionPipeline39from diffusers.utils import load_image40 41pipe = DiffusionPipeline.from_pretrained("Qwen/Qwen-Image-Edit-2509")42pipe.load_lora_weights("dx8152/Relight")43 44prompt = "Turn this cat into a dog"45input_image = load_image("https://huggingface.co/datasets/huggingface/documentation-images/resolve/main/diffusers/cat.png")46 47image = pipe(image=input_image, prompt=prompt).images[0]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='dx8152_Relight_2.txt',58        repo_id='model-metadata/code_execution_files',59        path_in_repo='dx8152_Relight_2.txt',60        repo_type='dataset',61    )62