kailashahirwar commited on
Commit
615de65
·
1 Parent(s): bcf59c3

names changes

Browse files
Files changed (1) hide show
  1. app.py +5 -5
app.py CHANGED
@@ -9,7 +9,7 @@ import gradio as gr
9
  import preprocess
10
  from huggingface_hub import login
11
 
12
- def extract_garment(input_img, cls):
13
  print(input_img, type(input_img), cls)
14
 
15
  input_dir = "input_image"
@@ -51,10 +51,10 @@ with gr.Blocks(css=css) as demo:
51
  with gr.Row():
52
  with gr.Column():
53
  input_image = gr.Image(label="Input Image", type='pil', height="400px", show_label=True)
54
- dropdown = gr.Dropdown(["upper", "lower", "dress"], value="upper", label="Extract garment",
55
- info="Select the garment type you wish to extract!")
56
 
57
- output_image = gr.Image(label="Extracted garment", type='pil', height="400px", show_label=True,
58
  show_download_button=True)
59
 
60
  with gr.Row():
@@ -63,7 +63,7 @@ with gr.Blocks(css=css) as demo:
63
 
64
  gr.on(
65
  triggers=[submit_button.click],
66
- fn=extract_garment,
67
  inputs=[input_image, dropdown],
68
  outputs=[output_image]
69
  )
 
9
  import preprocess
10
  from huggingface_hub import login
11
 
12
+ def extract_clothes(input_img, cls):
13
  print(input_img, type(input_img), cls)
14
 
15
  input_dir = "input_image"
 
51
  with gr.Row():
52
  with gr.Column():
53
  input_image = gr.Image(label="Input Image", type='pil', height="400px", show_label=True)
54
+ dropdown = gr.Dropdown(["upper", "lower", "dress"], value="upper", label="Extract clothes",
55
+ info="Select the clothes type you wish to extract!")
56
 
57
+ output_image = gr.Image(label="Extracted clothes", type='pil', height="400px", show_label=True,
58
  show_download_button=True)
59
 
60
  with gr.Row():
 
63
 
64
  gr.on(
65
  triggers=[submit_button.click],
66
+ fn=extract_clothes,
67
  inputs=[input_image, dropdown],
68
  outputs=[output_image]
69
  )