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)