paralym commited on
Commit
f362a76
·
verified ·
1 Parent(s): 9e1e5eb

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -239,6 +239,7 @@ def add_message(history, message):
239
  our_chatbot = InferenceDemo(
240
  args, model_path, tokenizer, model, image_processor, context_len
241
  )
 
242
 
243
  if len(message["files"]) <= 1:
244
  for x in message["files"]:
@@ -249,6 +250,7 @@ def add_message(history, message):
249
  our_chatbot = InferenceDemo(
250
  args, model_path, tokenizer, model, image_processor, context_len
251
  )
 
252
  for x in message["files"]:
253
  history.append(((x,), None))
254
  chat_image_num += 1
@@ -258,7 +260,6 @@ def add_message(history, message):
258
  else:
259
  for x in message["files"]:
260
  history.append(((x,), None))
261
- chat_image_num += len(message["files"])
262
  if message["text"] is not None:
263
  history.append((message["text"], None))
264
 
 
239
  our_chatbot = InferenceDemo(
240
  args, model_path, tokenizer, model, image_processor, context_len
241
  )
242
+ chat_image_num = 0
243
 
244
  if len(message["files"]) <= 1:
245
  for x in message["files"]:
 
250
  our_chatbot = InferenceDemo(
251
  args, model_path, tokenizer, model, image_processor, context_len
252
  )
253
+ chat_image_num = 0
254
  for x in message["files"]:
255
  history.append(((x,), None))
256
  chat_image_num += 1
 
260
  else:
261
  for x in message["files"]:
262
  history.append(((x,), None))
 
263
  if message["text"] is not None:
264
  history.append((message["text"], None))
265