gitglubber commited on
Commit
7a34e42
·
verified ·
1 Parent(s): c90c23b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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=1024
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("# Qwen Chatbot")
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")