Customise user agent header

replicate · Jun 28, 2024 · 82f320f · 82f320f
1 parent d949de6
commit 82f320f
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/node.py b/node.py
@@ -1,6 +1,5 @@
 import os
 import json
-import replicate
 import requests
 from PIL import Image
 from io import BytesIO
@@ -9,13 +8,16 @@
 import torch
 import base64
 import time
+from replicate.client import Client
 from .schema_to_node import (
     schema_to_comfyui_input_types,
     get_return_type,
     name_and_version,
     inputs_that_need_arrays,
 )
 
+replicate = Client(headers={"User-Agent": "comfyui-replicate/1.0.1"})
+
 
 def create_comfyui_node(schema):
     replicate_model, node_name = name_and_version(schema)