YC-Chen commited on
Commit
6130bd5
1 Parent(s): fc1ff08

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -117,11 +117,11 @@ with gr.Blocks() as demo:
117
  data = {
118
  "model": MODEL_NAME,
119
  "prompt": str(message),
120
- "temperature": temperature,
121
- "n": top_k,
122
- "max_tokens": max_new_tokens,
123
  "stop": "",
124
- "top_p": top_p,
125
  "logprobs": 0,
126
  "echo": False,
127
  "presence_penalty": PRESENCE_PENALTY,
 
117
  data = {
118
  "model": MODEL_NAME,
119
  "prompt": str(message),
120
+ "temperature": float(temperature),
121
+ "n": int(top_k),
122
+ "max_tokens": int(max_new_tokens),
123
  "stop": "",
124
+ "top_p": float(top_p),
125
  "logprobs": 0,
126
  "echo": False,
127
  "presence_penalty": PRESENCE_PENALTY,