Spaces:
Running
on
T4
Some suggestions for the demo!
Very cool demo! Just a few suggestions:
(1) I was getting an OOM error whenever I tried it -- you might need to upgrade to GPU?
(2) Can you set cache_examples = True to make it faster to see the results on the provided examples?
(3) Perhaps the gr.AnnotatedImage component would provide for a nicer output: https://gradio.app/docs/#annotatedimage
Also, if you cite your paper http://arxiv.org/abs/2306.12156 in the README, it will show up in the associated paper page: https://huggingface.co/papers/2306.12156
Thanks for your suggestion. I have upgraded to GPU and to show an example as "cache_examples = True". The queue and more cache examples will come soon.
Currently working with this demo (locally) adding some features (i.e. box prompt, mask prompt, show segment or show mask generated, load one input file and use for all prompts, etc.
Your demo and working code is very nice and much appreciated.