Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -43,10 +43,10 @@ class GenerateRequest(BaseModel):
|
|
43 |
input_text: str = ""
|
44 |
task_type: str
|
45 |
temperature: float = 1.0
|
46 |
-
max_new_tokens: int = 3
|
47 |
stream: bool = True
|
48 |
top_p: float = 1.0
|
49 |
-
top_k: int = 50
|
50 |
repetition_penalty: float = 1.0
|
51 |
num_return_sequences: int = 1
|
52 |
do_sample: bool = True
|
|
|
43 |
input_text: str = ""
|
44 |
task_type: str
|
45 |
temperature: float = 1.0
|
46 |
+
max_new_tokens: int = 3
|
47 |
stream: bool = True
|
48 |
top_p: float = 1.0
|
49 |
+
top_k: int = 50 # Changed back to 50
|
50 |
repetition_penalty: float = 1.0
|
51 |
num_return_sequences: int = 1
|
52 |
do_sample: bool = True
|