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    import os15    from huggingface_hub import InferenceClient16    17    client = InferenceClient(18        provider="auto",19        api_key=os.environ["HF_TOKEN"],20    )21    22    result = client.feature_extraction(23        "Today is a sunny day and I will get some ice cream.",24        model="ai-forever/FRIDA",25    )26    with open('ai-forever_FRIDA_2.txt', 'w', encoding='utf-8') as f:27        f.write('Everything was good in ai-forever_FRIDA_2.txt')28except Exception as e:29    import os30    from slack_sdk import WebClient31    client = WebClient(token=os.environ['SLACK_TOKEN'])32    client.chat_postMessage(33        channel='#exp-slack-alerts',34        text='Problem in <https://huggingface.co/datasets/model-metadata/code_execution_files/blob/main/ai-forever_FRIDA_2.txt|ai-forever_FRIDA_2.txt>',35    )36 37    with open('ai-forever_FRIDA_2.txt', 'a', encoding='utf-8') as f:38        import traceback39        f.write('''```CODE: 40import os41from huggingface_hub import InferenceClient42 43client = InferenceClient(44    provider="auto",45    api_key=os.environ["HF_TOKEN"],46)47 48result = client.feature_extraction(49    "Today is a sunny day and I will get some ice cream.",50    model="ai-forever/FRIDA",51)52```53 54ERROR: 55''')56        traceback.print_exc(file=f)57    58finally:59    from huggingface_hub import upload_file60    upload_file(61        path_or_fileobj='ai-forever_FRIDA_2.txt',62        repo_id='model-metadata/code_execution_files',63        path_in_repo='ai-forever_FRIDA_2.txt',64        repo_type='dataset',65    )66