Spaces:
Running
Running
| license: apache-2.0 | |
| title: VTON DEMO | |
| sdk: streamlit | |
| colorFrom: blue | |
| colorTo: blue | |
| pinned: true | |
| short_description: Virtual TryON Demo | |
| sdk_version: 1.42.2 | |
| # Virtual Try-On Application | |
| This application uses the IDM-VTON model via Replicate API to perform virtual try-on of garments on human images. | |
| ## Requirements | |
| - Python 3.6+ | |
| - `replicate` Python package | |
| ## Installation | |
| 1. Install the required package: | |
| ```bash | |
| pip install replicate | |
| ``` | |
| 2. Set up your Replicate API token as an environment variable: | |
| ```bash | |
| export REPLICATE_API_TOKEN='your_token_here' | |
| ``` | |
| You can get your API token from [Replicate](https://replicate.com/account/api-tokens) after creating an account. | |
| ## Usage | |
| Run the script with: | |
| ```bash | |
| python vton.py | |
| ``` | |
| By default, the script uses example images. To use your own images, modify the `garm_img`, `human_img`, and `garment_des` variables in the script. | |
| ### Function Usage | |
| You can also import the function in your own Python code: | |
| ```python | |
| from vton import virtual_try_on | |
| output_path = virtual_try_on( | |
| garm_img_url="URL_TO_GARMENT_IMAGE", | |
| human_img_url="URL_TO_HUMAN_IMAGE", | |
| garment_des="Description of the garment" | |
| ) | |
| ``` | |
| ## Output | |
| The script will save the generated try-on image as `output.jpg` in the current directory. | |
| ## Model Information | |
| This application uses the IDM-VTON model created by cuuupid on Replicate. The model performs virtual try-on by overlaying garments onto human images based on the provided description. |