wybertwang commited on
Commit
7651146
·
1 Parent(s): 23aff00

Fix run text refiner twice

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)]