HarmonyView update

app.py

@@ -24,14 +24,14 @@ Given a single-view image, HarmonyView is able to generate multiview-consistent
 
 Procedure: </br>
 **Step 1**. Upload an image or select an example.  ==> The foreground is masked out by SAM and we crop it as inputs. </br>
-**Step 2**. Select "Elevation angle "and click "Run generation". ==> Generate multiview images. The **Elevation angle** is the elevation of the input image. (This costs about 45s.) </br>
+**Step 2**. Select "Elevation angle "and click "Run generation". ==> Generate multiview images. The **Elevation angle** is the elevation of the Input image. (This costs about 45s.) </br>
 You may adjust the **Crop size** and **Elevation angle** to get a better result! <br>
 To reconstruct a NeRF or a 3D mesh from the generated images, please refer to our [github repository](https://github.com/byeongjun-park/HarmonyView). <br>
 We have heavily borrowed codes from [Syncdreamer](https://huggingface.co/spaces/liuyuan-pal/SyncDreamer), which is an our strong baseline. 
 '''
 _USER_GUIDE0 = "Step1: Please upload an image in the block above (or choose an example shown in the left)."
 # _USER_GUIDE1 = "Step1: Please select a **Crop size** and click **Crop it**."
-_USER_GUIDE2 = "Step2: Please choose a **Elevation angle** and click **Run Generate**. The **Elevation angle** is the elevation of the input image. This costs about 45s."
+_USER_GUIDE2 = "Step2: Please choose a **Elevation angle** and click **Run Generate**. The **Elevation angle** is the elevation of the Input image. This costs about 45s."
 _USER_GUIDE3 = "Generated multiview images are shown below! (You may adjust the **Crop size** and **Elevation angle** to get a better result!)"
 
 others = '''**Step 1**. Select "Crop size" and click "Crop it". ==> The foreground object is centered and resized. </br>'''
@@ -66,7 +66,7 @@ class BackgroundRemoval:
         return image
 
 def resize_inputs(original_image, sam_image, crop_size, background_removal):
-    image_input = original_image if background_removal == "Input Image" else sam_image
+    image_input = original_image if background_removal == "Input image" else sam_image
     if image_input is None: return None
     alpha_np = np.asarray(image_input)[:, :, 3]
     coords = np.stack(np.nonzero(alpha_np), 1)[:, (1, 0)]
@@ -180,20 +180,20 @@ def run_demo():
 
     # NOTE: Examples must match inputs
     examples_full = [
-        ['hf_demo/examples/monkey.png',30,200,"SAM Output"],
-        ['hf_demo/examples/cat.png',30,200,"SAM Output"],
-        ['hf_demo/examples/crab.png',30,200,"SAM Output"],
-        ['hf_demo/examples/elephant.png',30,200,"SAM Output"],
-        ['hf_demo/examples/flower.png',0,200,"SAM Output"],
-        ['hf_demo/examples/forest.png',30,200,"SAM Output"],
-        ['hf_demo/examples/teapot.png',20,200,"SAM Output"],
-        ['hf_demo/examples/basket.png',30,200,"SAM Output"],
+        ['hf_demo/examples/monkey.png',30,200,"SAM output"],
+        ['hf_demo/examples/cat.png',30,200,"SAM output"],
+        ['hf_demo/examples/crab.png',30,200,"SAM output"],
+        ['hf_demo/examples/elephant.png',30,200,"SAM output"],
+        ['hf_demo/examples/flower.png',0,200,"SAM output"],
+        ['hf_demo/examples/forest.png',30,200,"SAM output"],
+        ['hf_demo/examples/teapot.png',20,200,"SAM output"],
+        ['hf_demo/examples/basket.png',30,200,"SAM output"],
     ]
 
     image_block = gr.Image(type='pil', image_mode='RGBA', height=256, label='Input image', tool=None, interactive=True)
-    elevation = gr.Slider(-10, 40, 30, step=5, label='Elevation angle of the input image', interactive=True)
+    elevation = gr.Slider(-10, 40, 30, step=5, label='Elevation angle', interactive=True)
     crop_size = gr.Slider(120, 240, 200, step=10, label='Crop size', interactive=True)
-    background_removal = gr.Radio(["Input Image", "SAM Output"], value=["SAM Output"], label="Input to HarmonyView", info="Which image do you want for the input to HarmonyView?")
+    background_removal = gr.Radio(["Input image", "SAM output"], value=["SAM output"], label="Input to HarmonyView", info="Which image do you want for the input to HarmonyView?")
 
     # Compose demo layout & data flow.
     with gr.Blocks(title=_TITLE, css="hf_demo/style.css") as demo:
@@ -231,7 +231,7 @@ def run_demo():
                 input_block = gr.Image(type='pil', image_mode='RGBA', label="Input to HarmonyView", height=256, interactive=False)
                 background_removal.render()
                 with gr.Accordion('Advanced options', open=False):
-                    cfg_scale_1 = gr.Slider(1.0, 5.0, 2.0, step=0.1, label='Classifier free guidance 1', info='How consistent to be with the input image', interactive=True)
+                    cfg_scale_1 = gr.Slider(1.0, 5.0, 2.0, step=0.1, label='Classifier free guidance 1', info='How consistent to be with the Input image', interactive=True)
                     cfg_scale_2 = gr.Slider(0.5, 1.5, 1.0, step=0.1, label='Classifier free guidance 2', info='How diverse a novel view to create', interactive=True)
                     seed = gr.Number(6033, label='Random seed', interactive=True)
                 run_btn = gr.Button('Run generation', variant='primary', interactive=True)