Spaces:
Running
Running
Update app.py
Browse files
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=
|
| 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(
|