yizhangliu
commited on
Commit
•
dea5caa
1
Parent(s):
c72b4b1
Update app.py
Browse files
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(
|
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_pil, 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 |
|