wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-05-07 07:42:32 +08:00
Code Issues Actions 18 Packages Projects Releases Wiki Activity

Switch to urllib to avoid extra dependency

Browse Source
This commit is contained in:
City 2023-04-04 21:35:09 +02:00
parent df03f38556
commit f981efed75
1 changed files with 3 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
nodes.py
View File

@ -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,]