Discard response object after use

This commit is contained in:
City 2023-04-04 22:20:00 +02:00
parent 668d6382d8
commit 5ae97051bc

View File

@ -21,8 +21,8 @@ class LoadImageUrl:
CATEGORY = "image" CATEGORY = "image"
def load_image(self, url): def load_image(self, url):
r = request.urlopen(url) with request.urlopen(url) as r:
i = Image.open(r) i = Image.open(r)
image = i.convert("RGB") image = i.convert("RGB")
image = np.array(image).astype(np.float32) / 255.0 image = np.array(image).astype(np.float32) / 255.0
image = torch.from_numpy(image)[None,] image = torch.from_numpy(image)[None,]