Skip to content

H
Hydra Node Http
Commit fb7c5f1b authored by kurumuz's avatar kurumuz
Browse files
Options

fix

parent 96ac3de2
Hide whitespace changes
Inline Side-by-side
Showing with 5 additions and 1 deletion
main.py
View file @ fb7c5f1b
...@@ -73,6 +73,10 @@ class Masker(TypedDict): ...@@ -73,6 +73,10 @@ class Masker(TypedDict):
seed: int seed: int
mask: str mask: str
class Tags(TypedDict):
tag: str
count: int
class GenerationRequest(BaseModel): class GenerationRequest(BaseModel):
prompt: str prompt: str
image: str = None image: str = None
...@@ -104,7 +108,7 @@ class TextRequest(BaseModel): ...@@ -104,7 +108,7 @@ class TextRequest(BaseModel):
prompt: str prompt: str
class TagOutput(BaseModel): class TagOutput(BaseModel):
tags: List[List[str, int]] tags: List[Tags]
class TextOutput(BaseModel): class TextOutput(BaseModel):
is_safe: str is_safe: str
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment