Update app.py

app.py

@@ -539,7 +539,14 @@ def run_anything_task(input_image, text_prompt, task_type, inpaint_prompt, box_t
 
     if (task_type == 'Kosmos-2'):
         global kosmos_model, kosmos_processor
-        kosmos_image, kosmos_text, kosmos_entities = kosmos_generate_predictions(input_image, kosmos_input, kosmos_model, kosmos_processor)
+        if isinstance(input_image, dict):
+            image_pil, image = load_image(input_image['image'].convert("RGB"))
+            input_img = input_image['image']
+        else:
+            image_pil, image = load_image(input_image.convert("RGB"))
+            input_img = input_image
+            
+        kosmos_image, kosmos_text, kosmos_entities = kosmos_generate_predictions(input_img, kosmos_input, kosmos_model, kosmos_processor)
         run_task_time, time_cost_str = get_time_cost(run_task_time, time_cost_str)
         return None, None, time_cost_str, kosmos_image, gr.Textbox.update(visible=(time_cost_str !='')), kosmos_text, kosmos_entities
 
@@ -902,7 +909,7 @@ if __name__ == "__main__":
             DESCRIPTION += f'RAM from [RelateAnything](https://github.com/Luodian/RelateAnything). <br>'
         DESCRIPTION += f'Remove(cleaner) from [lama-cleaner](https://github.com/Sanster/lama-cleaner). <br>'
         if kosmos_enable:
-            DESCRIPTION += f'Kosmos-2 from [Kosmos-2](https://huggingface.co/spaces/ydshieh/Kosmos-2). <br>'
+            DESCRIPTION += f'Kosmos-2 from [Kosmos-2](https://github.com/microsoft/unilm/tree/master/kosmos-2). <br>'
         DESCRIPTION += f'Thanks for their excellent work.'
         DESCRIPTION += f'<p>For faster inference without waiting in queue, you may duplicate the space and upgrade to GPU in settings. \
                         <a href="https://huggingface.co/spaces/yizhangliu/Grounded-Segment-Anything?duplicate=true"><img style="display: inline; margin-top: 0em; margin-bottom: 0em" src="https://bit.ly/3gLdBN6" alt="Duplicate Space" /></a></p>'