yizhangliu commited on
Commit
c72b4b1
1 Parent(s): 27d8fa3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -546,7 +546,7 @@ def run_anything_task(input_image, text_prompt, task_type, inpaint_prompt, box_t
546
  image_pil, image = load_image(input_image.convert("RGB"))
547
  input_img = input_image
548
 
549
- kosmos_image, kosmos_text, kosmos_entities = kosmos_generate_predictions(input_img, kosmos_input, kosmos_model, kosmos_processor)
550
  run_task_time, time_cost_str = get_time_cost(run_task_time, time_cost_str)
551
  return None, None, time_cost_str, kosmos_image, gr.Textbox.update(visible=(time_cost_str !='')), kosmos_text, kosmos_entities
552
 
 
546
  image_pil, image = load_image(input_image.convert("RGB"))
547
  input_img = input_image
548
 
549
+ kosmos_image, kosmos_text, kosmos_entities = kosmos_generate_predictions(image, kosmos_input, kosmos_model, kosmos_processor)
550
  run_task_time, time_cost_str = get_time_cost(run_task_time, time_cost_str)
551
  return None, None, time_cost_str, kosmos_image, gr.Textbox.update(visible=(time_cost_str !='')), kosmos_text, kosmos_entities
552