Stable-X commited on
Commit
dcb1657
1 Parent(s): db811c0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -8
app.py CHANGED
@@ -97,7 +97,7 @@ def process_image(
97
  print(f"Processing image {name_base}{name_ext}")
98
 
99
  path_output_dir = tempfile.mkdtemp()
100
- path_out_png = os.path.join(path_output_dir, f"{name_base}_diffuse.png")
101
  input_image = Image.open(path_input)
102
  input_image = resize_image(input_image, default_image_processing_resolution)
103
 
@@ -143,7 +143,7 @@ def process_video(
143
  print(f"Processing video {name_base}{name_ext}")
144
 
145
  path_output_dir = tempfile.mkdtemp()
146
- path_out_vis = os.path.join(path_output_dir, f"{name_base}_diffuse_colored.mp4")
147
 
148
  init_latents = None
149
  reader, writer = None, None
@@ -212,7 +212,7 @@ def run_demo_server(pipe):
212
 
213
  with gr.Blocks(
214
  theme=gradio_theme,
215
- title="Stable Diffuse Estimation",
216
  css="""
217
  #download {
218
  height: 118px;
@@ -256,7 +256,7 @@ def run_demo_server(pipe):
256
  ) as demo:
257
  gr.Markdown(
258
  """
259
- # StableDiffuse: Removing Reflections from Textured Surfaces in a Single Image
260
  <p align="center">
261
  """
262
  )
@@ -271,12 +271,12 @@ def run_demo_server(pipe):
271
  )
272
  with gr.Row():
273
  image_submit_btn = gr.Button(
274
- value="Compute Diffuse", variant="primary"
275
  )
276
  image_reset_btn = gr.Button(value="Reset")
277
  with gr.Column():
278
  image_output_slider = ImageSlider(
279
- label="Diffuse outputs",
280
  type="filepath",
281
  show_download_button=True,
282
  show_share_button=True,
@@ -306,7 +306,7 @@ def run_demo_server(pipe):
306
  )
307
  with gr.Row():
308
  video_submit_btn = gr.Button(
309
- value="Compute Diffuse", variant="primary"
310
  )
311
  video_reset_btn = gr.Button(value="Reset")
312
  with gr.Column():
@@ -320,7 +320,7 @@ def run_demo_server(pipe):
320
  position=0.25,
321
  )
322
  video_output_files = gr.Files(
323
- label="Diffuse outputs",
324
  elem_id="download",
325
  interactive=False,
326
  )
 
97
  print(f"Processing image {name_base}{name_ext}")
98
 
99
  path_output_dir = tempfile.mkdtemp()
100
+ path_out_png = os.path.join(path_output_dir, f"{name_base}_delight.png")
101
  input_image = Image.open(path_input)
102
  input_image = resize_image(input_image, default_image_processing_resolution)
103
 
 
143
  print(f"Processing video {name_base}{name_ext}")
144
 
145
  path_output_dir = tempfile.mkdtemp()
146
+ path_out_vis = os.path.join(path_output_dir, f"{name_base}_delight_colored.mp4")
147
 
148
  init_latents = None
149
  reader, writer = None, None
 
212
 
213
  with gr.Blocks(
214
  theme=gradio_theme,
215
+ title="Stable Delight Estimation",
216
  css="""
217
  #download {
218
  height: 118px;
 
256
  ) as demo:
257
  gr.Markdown(
258
  """
259
+ # StableDelight: Removing Reflections from Textured Surfaces in a Single Image
260
  <p align="center">
261
  """
262
  )
 
271
  )
272
  with gr.Row():
273
  image_submit_btn = gr.Button(
274
+ value="Delighting", variant="primary"
275
  )
276
  image_reset_btn = gr.Button(value="Reset")
277
  with gr.Column():
278
  image_output_slider = ImageSlider(
279
+ label="Delight outputs",
280
  type="filepath",
281
  show_download_button=True,
282
  show_share_button=True,
 
306
  )
307
  with gr.Row():
308
  video_submit_btn = gr.Button(
309
+ value="Delighting", variant="primary"
310
  )
311
  video_reset_btn = gr.Button(value="Reset")
312
  with gr.Column():
 
320
  position=0.25,
321
  )
322
  video_output_files = gr.Files(
323
+ label="Delight outputs",
324
  elem_id="download",
325
  interactive=False,
326
  )