nsfw mark api

api/models/GenerationRequest.py

@@ -23,6 +23,10 @@ class GenerationRequest(BaseModel):
     count: int = Field(default=1, ge=1, le=10)
 
 
+class NsfwRequest(BaseModel):
+    is_nsfw: bool
+
+
 class GenerationsResponse(BaseModel):
     generations: List["GenerationResponse"]
     total_count: int
@@ -69,4 +73,4 @@ class PromptRequest(BaseModel):
 
 
 class PromptResponse(BaseModel):
-    prompt: str
+    prompt: str