Tonic commited on
Commit
76a70ea
·
1 Parent(s): 3917857

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -6
app.py CHANGED
@@ -122,8 +122,8 @@ def _launch_demo(args, model, tokenizer):
122
  processed_history.append(item)
123
  else:
124
  print("Invalid history item:", item)
125
- history = process_history_for_model(processed_history)
126
- response, history = model.chat(tokenizer, query=query, history=history)
127
  image = tokenizer.draw_bbox_on_latest_picture(response, history)
128
  if image is not None:
129
  image_path = save_image(image)
@@ -175,11 +175,11 @@ def _launch_demo(args, model, tokenizer):
175
  processed_history = []
176
  for query, response in task_history:
177
  if isinstance(query, tuple):
178
- processed_history.append({'image': query[0]})
179
  else:
180
- processed_history.append({'text': query})
181
- if response:
182
- processed_history.append({'text': response})
183
  return processed_history
184
 
185
  def reset_state(task_history):
 
122
  processed_history.append(item)
123
  else:
124
  print("Invalid history item:", item)
125
+ processed_history = process_history_for_model(task_history)
126
+ response, history = model.chat(tokenizer, query=query, history=processed_history)
127
  image = tokenizer.draw_bbox_on_latest_picture(response, history)
128
  if image is not None:
129
  image_path = save_image(image)
 
175
  processed_history = []
176
  for query, response in task_history:
177
  if isinstance(query, tuple):
178
+ query = {'image': query[0]}
179
  else:
180
+ query = {'text': query}
181
+ response = response or ""
182
+ processed_history.append((query, response))
183
  return processed_history
184
 
185
  def reset_state(task_history):