Update app.py
Browse files
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 |
-
|
126 |
-
response, history = model.chat(tokenizer, query=query, 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 |
-
|
179 |
else:
|
180 |
-
|
181 |
-
|
182 |
-
|
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):
|