Hjgugugjhuhjggg commited on
Commit
8b0fc48
·
verified ·
1 Parent(s): 5166951

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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