From 82f320f601207c70c29864ceb12ac2e4b2206070 Mon Sep 17 00:00:00 2001
From: Paul <fofr@users.noreply.github.com>
Date: Fri, 28 Jun 2024 13:11:12 +0000
Subject: [PATCH] Customise user agent header

---
 node.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/node.py b/node.py
index 6874479..2d59372 100644
--- 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,6 +8,7 @@ from torchvision import transforms
 import torch
 import base64
 import time
+from replicate.client import Client
 from .schema_to_node import (
     schema_to_comfyui_input_types,
     get_return_type,
@@ -16,6 +16,8 @@ from .schema_to_node import (
     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)