fix

fb7c5f1b · kurumuz · 96ac3de2 · fb7c5f1b
Commit fb7c5f1b authored Sep 26, 2022 by kurumuz
Hide whitespace changes
Inline Side-by-side

Showing with 5 additions and 1 deletion

main.py main.py +5 -1

No files found.
--- a/main.py
+++ b/main.py
@@ -73,6 +73,10 @@ class Masker(TypedDict):
    seed: int
    mask: str

+class Tags(TypedDict):
+    tag: str
+    count: int
+
 class GenerationRequest(BaseModel):
    prompt: str
    image: str = None
@@ -104,7 +108,7 @@ class TextRequest(BaseModel):
    prompt: str

 class TagOutput(BaseModel):
-    tags: List[List[str, int]]
+    tags: List[Tags]

 class TextOutput(BaseModel):
    is_safe: str