cristinaimprota commited on
Commit
e733d81
·
verified ·
1 Parent(s): 622efd2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -66,7 +66,7 @@ def generate_code(instruction, chat_history, is_first_time):
66
  with torch.no_grad():
67
  outputs = model.generate(
68
  **inputs,
69
- max_new_tokens=512,
70
  do_sample=False,
71
  pad_token_id=stop_id,
72
  eos_token_id=stop_id,
@@ -118,7 +118,7 @@ with gr.Blocks(title="Python Code Generator") as demo:
118
  chat_output = gr.Chatbot(
119
  label="Conversation",
120
  height=500,
121
- type=messages,
122
  )
123
 
124
  submit_btn.click(
 
66
  with torch.no_grad():
67
  outputs = model.generate(
68
  **inputs,
69
+ max_new_tokens=256,
70
  do_sample=False,
71
  pad_token_id=stop_id,
72
  eos_token_id=stop_id,
 
118
  chat_output = gr.Chatbot(
119
  label="Conversation",
120
  height=500,
121
+ type="messages",
122
  )
123
 
124
  submit_btn.click(