Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -38,7 +38,7 @@ def generate_response(chat_history):
|
|
| 38 |
# Generate text
|
| 39 |
generated_ids = model.generate(
|
| 40 |
**model_inputs,
|
| 41 |
-
max_new_tokens=
|
| 42 |
)
|
| 43 |
output_ids = generated_ids[0][len(model_inputs.input_ids[0]):].tolist()
|
| 44 |
|
|
@@ -47,7 +47,7 @@ def generate_response(chat_history):
|
|
| 47 |
|
| 48 |
# --- Gradio Interface ---
|
| 49 |
with gr.Blocks() as demo:
|
| 50 |
-
gr.Markdown("#
|
| 51 |
# We use a state object to store the system message, though it's constant here
|
| 52 |
chatbot = gr.Chatbot()
|
| 53 |
msg = gr.Textbox(label="Input")
|
|
|
|
| 38 |
# Generate text
|
| 39 |
generated_ids = model.generate(
|
| 40 |
**model_inputs,
|
| 41 |
+
max_new_tokens=8192
|
| 42 |
)
|
| 43 |
output_ids = generated_ids[0][len(model_inputs.input_ids[0]):].tolist()
|
| 44 |
|
|
|
|
| 47 |
|
| 48 |
# --- Gradio Interface ---
|
| 49 |
with gr.Blocks() as demo:
|
| 50 |
+
gr.Markdown("# Slide Chat Bot")
|
| 51 |
# We use a state object to store the system message, though it's constant here
|
| 52 |
chatbot = gr.Chatbot()
|
| 53 |
msg = gr.Textbox(label="Input")
|