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("Qwen/Qwen-Image-Edit-2509")25    pipe.load_lora_weights("dx8152/Fusion_lora")26    27    prompt = "Turn this cat into a dog"28    input_image = load_image("https://huggingface.co/datasets/huggingface/documentation-images/resolve/main/diffusers/cat.png")29    30    image = pipe(image=input_image, prompt=prompt).images[0]31    with open('dx8152_Fusion_lora_0.txt', 'w', encoding='utf-8') as f:32        f.write('Everything was good in dx8152_Fusion_lora_0.txt')33except Exception as e:34    import os35    from slack_sdk import WebClient36    client = WebClient(token=os.environ['SLACK_TOKEN'])37    client.chat_postMessage(38        channel='#hub-model-metadata-snippets-sprint',39        text='Problem in <https://huggingface.co/datasets/model-metadata/code_execution_files/blob/main/dx8152_Fusion_lora_0.txt|dx8152_Fusion_lora_0.txt>',40    )41 42    with open('dx8152_Fusion_lora_0.txt', 'a', encoding='utf-8') as f:43        import traceback44        f.write('''```CODE: 45from diffusers import DiffusionPipeline46from diffusers.utils import load_image47 48pipe = DiffusionPipeline.from_pretrained("Qwen/Qwen-Image-Edit-2509")49pipe.load_lora_weights("dx8152/Fusion_lora")50 51prompt = "Turn this cat into a dog"52input_image = load_image("https://huggingface.co/datasets/huggingface/documentation-images/resolve/main/diffusers/cat.png")53 54image = pipe(image=input_image, prompt=prompt).images[0]55```56 57ERROR: 58''')59        traceback.print_exc(file=f)60    61finally:62    from huggingface_hub import upload_file63    upload_file(64        path_or_fileobj='dx8152_Fusion_lora_0.txt',65        repo_id='model-metadata/code_execution_files',66        path_in_repo='dx8152_Fusion_lora_0.txt',67        repo_type='dataset',68    )69