update app.py

app.py

@@ -698,8 +698,6 @@ def run_anything_task(input_image, text_prompt, task_type, inpaint_prompt, box_t
     logger.info(f'run_anything_task_[{file_temp}]_9_9_')
     return output_images, gr.Gallery.update(label='result images'), None, None, None
 
-kosmos_enable = False
-
 def change_radio_display(task_type, mask_source_radio):
     text_prompt_visible = True
     inpaint_prompt_visible = False
@@ -717,8 +715,7 @@ def change_radio_display(task_type, mask_source_radio):
             image_gallery_visible = False
             kosmos_input_visible = True
             kosmos_output_visible = True
-            kosmos_text_output_visible = True
-        
+            kosmos_text_output_visible = True        
 
     if task_type == "inpainting":
         inpaint_prompt_visible = True
@@ -754,7 +751,12 @@ def get_model_device(module):
     except Exception as e:
         return 'Error'
 
-task_types = ["detection", "segment", "inpainting", "remove", "relate anything"]
+ram_enable = True
+kosmos_enable = False
+
+task_types = ["detection", "segment", "inpainting", "remove"]
+if ram_enable:
+    task_types.append("relate anything")
 if kosmos_enable:
     task_types.append("Kosmos-2")
 
@@ -774,7 +776,8 @@ if __name__ == "__main__":
 
     load_sd_model()
     load_lama_cleaner_model()
-    load_ram_model()
+    if ram_enable:
+        load_ram_model()
     
     if os.environ.get('IS_MY_DEBUG') is None and kosmos_enable:
         kosmos_model, kosmos_processor = load_kosmos_model(device)
@@ -867,7 +870,8 @@ if __name__ == "__main__":
                             ])
 
         DESCRIPTION = f'### This demo from [Grounded-Segment-Anything](https://github.com/IDEA-Research/Grounded-Segment-Anything). <br>'
-        DESCRIPTION += f'RAM from [RelateAnything](https://github.com/Luodian/RelateAnything). <br>'
+        if ram_enable:
+            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>'

app_cli.py

@@ -77,6 +77,8 @@ sam_mask_generator = None
 sd_pipe = None
 lama_cleaner_model= None
 ram_model = None
+kosmos_model = None
+kosmos_processor = None
 
 def get_args():
     argparser = argparse.ArgumentParser()
@@ -116,6 +118,7 @@ if __name__ == '__main__':
                         remove_mode = "rectangle",   # ["segment", "rectangle"]
                         remove_mask_extend = "10", 
                         num_relation = 5,
+                        kosmos_input = None,
                         cleaner_size_limit = -1,
                         )
     if len(output_images) > 0: