| import os | |
| import gradio as gr | |
| from model import predict | |
| description = """ | |
| - This work is a part of the [DepthPro: Beyond Depth Estimation](https://github.com/geetu040/depthpro-beyond-depth) repository, which further explores this model's capabilities on: | |
| - Image Segmentation - Human Segmentation | |
| - Image Super Resolution - 384px to 1536px (4x Upscaling) | |
| - Image Super Resolution - 256px to 1024px (4x Upscaling) | |
| """ | |
| examples_dir = "assets/examples/" | |
| examples = [[os.path.join(examples_dir, filename)] for filename in os.listdir(examples_dir)] | |
| interface = gr.Interface( | |
| fn=predict, | |
| inputs=gr.Image(type="pil"), | |
| outputs=gr.Image(type="pil"), | |
| title="DepthPro: Segmentation Human", | |
| description=description, | |
| examples=examples, | |
| ) | |
| if __name__ == "__main__": | |
| interface.launch() | |