From f981efed75d092172aa2f9bbda295e86c1d7f586 Mon Sep 17 00:00:00 2001 From: City <125218114+city96@users.noreply.github.com> Date: Tue, 4 Apr 2023 21:35:09 +0200 Subject: [PATCH] Switch to urllib to avoid extra dependency --- nodes.py | 7 +++---- 1 file changed, 3 insertions(+), 4 deletions(-) diff --git a/nodes.py b/nodes.py index cf41aba31..bf850e7d2 100644 --- a/nodes.py +++ b/nodes.py @@ -10,7 +10,7 @@ import traceback from PIL import Image from PIL.PngImagePlugin import PngInfo import numpy as np -import requests +from urllib import request sys.path.insert(0, os.path.join(os.path.dirname(os.path.realpath(__file__)), "comfy")) @@ -954,9 +954,8 @@ class LoadImageUrl: CATEGORY = "image" def load_image(self, url): - r = requests.get(url, stream=True) - r.raise_for_status() - i = Image.open(r.raw) + r = request.urlopen(url) + i = Image.open(r) image = i.convert("RGB") image = np.array(image).astype(np.float32) / 255.0 image = torch.from_numpy(image)[None,]