VTON-PRJ / app.py
priyanaik21's picture
Create app.py
1ac01ce verified
raw
history blame
704 Bytes
import gradio as gr
with gr.Blocks() as app:
with gr.Row():
gr.Markdown(" Clothes Virtual Try ON ")
with gr.Row():
with gr.Column():
cloth_input = gr.Image(sources=['upload'], type="pil", label="Upload the Cloth Image")
with gr.Column():
model_input = gr.Image(sources=['upload'], type="pil", label="Upload the Human Image")
with gr.Row():
final_output = gr.Image(sources=['upload'], type="pil", label="Final Prediction")
with gr.Row():
submit_button = gr.Button("Submit")
submit_button.click(fn=run, inputs=[cloth_input, model_input],
outputs=[final_output])
app.launch(debug=True, share=True)