ttengwang commited on
Commit
35b6cee
·
2 Parent(s): 3421695 7651146

Merge branch 'main' of https://huggingface.co/spaces/wybertwang/Caption-Anything into main

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -173,7 +173,7 @@ def inference_seg_cap(image_input, point_prompt, click_mode, language, sentiment
173
  prompt = get_prompt(coordinate, click_state, click_mode)
174
  print('prompt: ', prompt, 'controls: ', controls)
175
 
176
- out = model.inference(image_input, prompt, controls)
177
  state = state + [(None, "Image point: {}, Input label: {}".format(prompt["input_point"], prompt["input_label"]))]
178
  # for k, v in out['generated_captions'].items():
179
  # state = state + [(f'{k}: {v}', None)]
 
173
  prompt = get_prompt(coordinate, click_state, click_mode)
174
  print('prompt: ', prompt, 'controls: ', controls)
175
 
176
+ out = model.inference(image_input, prompt, controls, disable_gpt=True)
177
  state = state + [(None, "Image point: {}, Input label: {}".format(prompt["input_point"], prompt["input_label"]))]
178
  # for k, v in out['generated_captions'].items():
179
  # state = state + [(f'{k}: {v}', None)]