junyangwang0410 commited on
Commit
6b356fd
·
verified ·
1 Parent(s): 681ff13

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -311,7 +311,7 @@ def chatbot(image, instruction, add_info, history, chat_log):
311
  chat_action = add_response("user", prompt_action, chat_action, [screenshot_som_file])
312
  output_action = inference_chat(chat_action, vl_model_version, API_url, token)
313
  output_for_save_this_step['action'] = output_action
314
- action_json = output_action
315
  thought = action_json['Thought']
316
  summary = action_json['Summary']
317
  action = action_json['Action']
 
311
  chat_action = add_response("user", prompt_action, chat_action, [screenshot_som_file])
312
  output_action = inference_chat(chat_action, vl_model_version, API_url, token)
313
  output_for_save_this_step['action'] = output_action
314
+ action_json = json.loads(output_action.split("```json")[-1].split("```")[0])
315
  thought = action_json['Thought']
316
  summary = action_json['Summary']
317
  action = action_json['Action']