hangzhang-nlp commited on
Commit
df5bd23
β€’
1 Parent(s): 1af558a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -128,7 +128,7 @@ def gradio_ask(user_message, chatbot, chat_state):
128
  def gradio_answer(chatbot, chat_state, img_list, num_beams, temperature):
129
  llm_message = chat.answer(conv=chat_state,
130
  img_list=img_list,
131
- num_beams=num_beams,
132
  temperature=temperature,
133
  max_new_tokens=300,
134
  max_length=2000)[0]
 
128
  def gradio_answer(chatbot, chat_state, img_list, num_beams, temperature):
129
  llm_message = chat.answer(conv=chat_state,
130
  img_list=img_list,
131
+ num_beams=1,
132
  temperature=temperature,
133
  max_new_tokens=300,
134
  max_length=2000)[0]