Update app.py
Browse files
app.py
CHANGED
@@ -23,7 +23,7 @@ class GenerateRequest(BaseModel):
|
|
23 |
input_text: str = ""
|
24 |
task_type: str
|
25 |
temperature: float = 1.0
|
26 |
-
max_new_tokens: int =
|
27 |
stream: bool = True
|
28 |
top_p: float = 1.0
|
29 |
top_k: int = 50
|
|
|
23 |
input_text: str = ""
|
24 |
task_type: str
|
25 |
temperature: float = 1.0
|
26 |
+
max_new_tokens: int = 2
|
27 |
stream: bool = True
|
28 |
top_p: float = 1.0
|
29 |
top_k: int = 50
|