Spaces:
Running
Running
Commit
·
b13f641
1
Parent(s):
c90fa16
Refactor GenerateRequest model; update emotional_tone field alias and improve field definitions for clarity
Browse files
main.py
CHANGED
|
@@ -110,11 +110,11 @@ class StyleMapper:
|
|
| 110 |
class GenerateRequest(BaseModel):
|
| 111 |
prompt: str
|
| 112 |
style: PoemStyle
|
| 113 |
-
emotional_tone: EmotionalTone
|
| 114 |
-
creative_style: float = Field(ge=0, le=100, alias="creativeStyle")
|
| 115 |
-
language_variety: float = Field(ge=0, le=1, alias="languageVariety")
|
| 116 |
length: Length
|
| 117 |
-
word_repetition: float = Field(ge=1, le=2, alias="wordRepetition")
|
| 118 |
|
| 119 |
@validator('creative_style')
|
| 120 |
def validate_creative_style(cls, v):
|
|
@@ -136,7 +136,7 @@ class GenerateRequest(BaseModel):
|
|
| 136 |
|
| 137 |
class Config:
|
| 138 |
allow_population_by_field_name = True
|
| 139 |
-
|
| 140 |
|
| 141 |
class ModelManager:
|
| 142 |
def __init__(self):
|
|
|
|
| 110 |
class GenerateRequest(BaseModel):
|
| 111 |
prompt: str
|
| 112 |
style: PoemStyle
|
| 113 |
+
emotional_tone: EmotionalTone = Field(alias="emotionalTone")
|
| 114 |
+
creative_style: float = Field(ge=0, le=100, alias="creativeStyle") # 0-100 slider
|
| 115 |
+
language_variety: float = Field(ge=0, le=1, alias="languageVariety") # 0-1 slider
|
| 116 |
length: Length
|
| 117 |
+
word_repetition: float = Field(ge=1, le=2, alias="wordRepetition") # 1-2 slider
|
| 118 |
|
| 119 |
@validator('creative_style')
|
| 120 |
def validate_creative_style(cls, v):
|
|
|
|
| 136 |
|
| 137 |
class Config:
|
| 138 |
allow_population_by_field_name = True
|
| 139 |
+
|
| 140 |
|
| 141 |
class ModelManager:
|
| 142 |
def __init__(self):
|