This commit is contained in:
thot-experiment 2025-04-23 11:39:12 -07:00
parent cf209aa456
commit 6fb5bb1431
2 changed files with 3 additions and 3 deletions

View File

@ -6,10 +6,10 @@ from __future__ import annotations
from datetime import datetime from datetime import datetime
from enum import Enum from enum import Enum
from typing import Any, Dict, List, Literal, Optional, Union from typing import Any, Dict, List, Optional, Union
from uuid import UUID from uuid import UUID
from pydantic import AnyUrl, BaseModel, Field, RootModel, confloat, conint, constr from pydantic import AnyUrl, BaseModel, Field, confloat, conint
class ComfyNode(BaseModel): class ComfyNode(BaseModel):
category: Optional[str] = Field( category: Optional[str] = Field(

View File

@ -46,7 +46,7 @@ def validate_and_cast_response (response):
if b64_data: if b64_data:
img_data = base64.b64decode(b64_data) img_data = base64.b64decode(b64_data)
img = Image.open(io.BytesIO(img_data)) img = Image.open(io.BytesIO(img_data))
elif image_url: elif image_url:
img_response = requests.get(image_url) img_response = requests.get(image_url)
if img_response.status_code != 200: if img_response.status_code != 200: