model

561ecafa · kurumuz · 7ae5e142 · 561ecafa
Commit 561ecafa authored Oct 01, 2022 by kurumuz
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 0 deletions

main.py main.py +1 -0

No files found.
--- a/main.py
+++ b/main.py
@@ -107,6 +107,7 @@ class GenerationRequest(BaseModel):

 class TextRequest(BaseModel):
    prompt: str
+    model: str

 class TagOutput(BaseModel):
    tags: List[Tags]