Yuliang commited on
Commit
b151252
1 Parent(s): 3df5b98

label update

Browse files
Files changed (1) hide show
  1. app.py +7 -7
app.py CHANGED
@@ -107,14 +107,14 @@ with gr.Blocks() as demo:
107
  with gr.Row():
108
  with gr.Column():
109
  seed = gr.inputs.Slider(
110
- 0, 100, step=1, default=0, label='Seed (For Image Generation)')
111
  psi = gr.inputs.Slider(
112
  0, 2, step=0.05, default=0.7, label='Truncation psi (For Image Generation)')
113
  radio_choice = gr.Radio(
114
  model_types, label='Method (For Reconstruction)', value='icon-filter')
115
  inp = gr.Image(type="filepath", label="Input Image")
116
  with gr.Row():
117
- btn_sample = gr.Button("Sample Image")
118
  btn_submit = gr.Button("Submit Image")
119
 
120
  gr.Examples(examples=examples,
@@ -123,7 +123,7 @@ with gr.Blocks() as demo:
123
  fn=generate_model,
124
  outputs=out_lst)
125
 
126
- out_vid = gr.Video(label="Image + Normal + Recon + Refined Recon")
127
  out_vid_download = gr.File(
128
  label="Download Video, welcome share on Twitter with #ICON")
129
 
@@ -131,12 +131,12 @@ with gr.Blocks() as demo:
131
  overlap_inp = gr.Image(
132
  type="filepath", label="Image Normal Overlap")
133
  out_final = gr.Model3D(
134
- clear_color=[0.0, 0.0, 0.0, 0.0], label="Refined Recon")
135
  out_final_download = gr.File(
136
- label="Download refined clothed human mesh")
137
  out_smpl = gr.Model3D(
138
- clear_color=[0.0, 0.0, 0.0, 0.0], label="SMPL")
139
- out_smpl_download = gr.File(label="Download SMPL mesh")
140
  out_smpl_npy_download = gr.File(label="Download SMPL params")
141
 
142
  out_lst = [out_smpl, out_smpl_download, out_smpl_npy_download,
 
107
  with gr.Row():
108
  with gr.Column():
109
  seed = gr.inputs.Slider(
110
+ 0, 1000, step=1, default=0, label='Seed (For Image Generation)')
111
  psi = gr.inputs.Slider(
112
  0, 2, step=0.05, default=0.7, label='Truncation psi (For Image Generation)')
113
  radio_choice = gr.Radio(
114
  model_types, label='Method (For Reconstruction)', value='icon-filter')
115
  inp = gr.Image(type="filepath", label="Input Image")
116
  with gr.Row():
117
+ btn_sample = gr.Button("Generate Image")
118
  btn_submit = gr.Button("Submit Image")
119
 
120
  gr.Examples(examples=examples,
 
123
  fn=generate_model,
124
  outputs=out_lst)
125
 
126
+ out_vid = gr.Video(label="Image + Normal + SMPL Body + Clothed Human")
127
  out_vid_download = gr.File(
128
  label="Download Video, welcome share on Twitter with #ICON")
129
 
 
131
  overlap_inp = gr.Image(
132
  type="filepath", label="Image Normal Overlap")
133
  out_final = gr.Model3D(
134
+ clear_color=[0.0, 0.0, 0.0, 0.0], label="Clothed human")
135
  out_final_download = gr.File(
136
+ label="Download clothed human mesh")
137
  out_smpl = gr.Model3D(
138
+ clear_color=[0.0, 0.0, 0.0, 0.0], label="SMPL body")
139
+ out_smpl_download = gr.File(label="Download SMPL body mesh")
140
  out_smpl_npy_download = gr.File(label="Download SMPL params")
141
 
142
  out_lst = [out_smpl, out_smpl_download, out_smpl_npy_download,