update app.py

app.py

@@ -237,7 +237,7 @@ groundingdino_model = load_model_hf(config_file, ckpt_repo_id, ckpt_filenmae)
 
 # initialize SAM
 logger.info(f"initialize SAM model...")
-sam_model = build_sam(checkpoint=sam_checkpoint).to(device)
+sam_model = build_sam(checkpoint=sam_checkpoint) # .to(device)
 sam_predictor = SamPredictor(sam_model)
 sam_mask_generator = SamAutomaticMaskGenerator(sam_model)
 
@@ -726,16 +726,18 @@ if __name__ == "__main__":
             with gr.Column():
                 # gsa_gallery = gr.Gallery(
                 #     label="result images", show_label=True, elem_id="gsa_gallery"
-                # ).style(grid=[2], full_width=True, full_height=True)
-                gallery = gr.Gallery(label="Your Result", show_label=True, elem_id="gallery").style(preview=True, columns=5, object_fit="scale-down")
-
+                #     ).style(grid=[2], full_width=True, full_height=True)
+                gallery = gr.Gallery(label="result images", show_label=True, elem_id="gallery")
+                    .style(preview=True, columns=5, object_fit="scale-down", full_width=True, full_height=True)
 
         run_button.click(fn=run_anything_task, inputs=[
                         input_image, text_prompt, task_type, inpaint_prompt, box_threshold, text_threshold, iou_threshold, inpaint_mode, mask_source_radio, remove_mode, remove_mask_extend, num_relation], outputs=[gallery, gallery])
         task_type.change(fn=change_radio_display, inputs=[task_type, mask_source_radio], outputs=[text_prompt, inpaint_prompt, mask_source_radio, num_relation])
         mask_source_radio.change(fn=change_radio_display, inputs=[task_type, mask_source_radio], outputs=[text_prompt, inpaint_prompt, mask_source_radio, num_relation])
 
-        DESCRIPTION = '### This demo from [Grounded-Segment-Anything](https://github.com/IDEA-Research/Grounded-Segment-Anything). Thanks for their excellent work.'
+        DESCRIPTION = '### This demo from [Grounded-Segment-Anything](https://github.com/IDEA-Research/Grounded-Segment-Anything). <br>'
+        DESCRIPTION += '### RAM from [RelateAnything](https://github.com/Luodian/RelateAnything). <br>'
+        DESCRIPTION += '### 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>'
         gr.Markdown(DESCRIPTION)