HarmonyView update

.idea/workspace.xml

@@ -17,6 +17,9 @@
   <component name="MarkdownSettingsMigration">
     <option name="stateVersion" value="1" />
   </component>
+  <component name="PerforceDirect.Settings">
+    <option name="CHARSET" value="없음" />
+  </component>
   <component name="ProjectColorInfo">{
   &quot;associatedIndex&quot;: 6
 }</component>
@@ -62,7 +65,7 @@
       <workItem from="1704106389325" duration="3000" />
       <workItem from="1707152826665" duration="2000" />
       <workItem from="1709009070011" duration="10000" />
-      <workItem from="1717834285162" duration="1496000" />
+      <workItem from="1717834285162" duration="1820000" />
     </task>
     <task id="LOCAL-00001" summary="error resolve">
       <option name="closed" value="true" />
@@ -264,7 +267,15 @@
       <option name="project" value="LOCAL" />
       <updated>1717835213390</updated>
     </task>
-    <option name="localTasksCounter" value="26" />
+    <task id="LOCAL-00026" summary="HarmonyView update">
+      <option name="closed" value="true" />
+      <created>1717835840224</created>
+      <option name="number" value="00026" />
+      <option name="presentableId" value="LOCAL-00026" />
+      <option name="project" value="LOCAL" />
+      <updated>1717835840224</updated>
+    </task>
+    <option name="localTasksCounter" value="27" />
     <servers />
   </component>
   <component name="Vcs.Log.Tabs.Properties">

README.md

@@ -4,7 +4,7 @@ emoji: 🚀
 colorFrom: indigo
 colorTo: pink
 sdk: gradio
-sdk_version: 4.35.0
+sdk_version: 4.36.0
 app_file: app.py
 pinned: false
 license: cc-by-sa-3.0

app.py

@@ -193,7 +193,7 @@ def run_demo():
         ['hf_demo/examples/teapot.png',20,200,"SAM output"],
     ]
 
-    image_block = gr.ImageEditor(type='pil', image_mode='RGBA', height=256, label='Input image', interactive=True)
+    image_block = gr.Image(type='pil', image_mode='RGBA', height=256, label='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"], label="Input to HarmonyView", info="Which image do you want for the input to HarmonyView?")
@@ -220,17 +220,17 @@ def run_demo():
             with gr.Column(scale=0.8):
                 image_block.render()
                 crop_size.render()
-                fig0 = gr.ImageEditor(value=Image.open('assets/crop_size.jpg'), type='pil', image_mode='RGB', height=256, show_label=False, tool=None, interactive=False)
+                fig0 = gr.Image(value=Image.open('assets/crop_size.jpg'), type='pil', image_mode='RGB', height=256, show_label=False, interactive=False)
 
 
             with gr.Column(scale=0.8):
-                sam_block = gr.ImageEditor(type='pil', image_mode='RGBA', label="SAM output", height=256, interactive=False)
+                sam_block = gr.Image(type='pil', image_mode='RGBA', label="SAM output", height=256, interactive=False)
                 # crop_btn = gr.Button('Crop it', variant='primary', interactive=True)
                 elevation.render()
-                fig1 = gr.ImageEditor(value=Image.open('assets/elevation.jpg'), type='pil', image_mode='RGB', height=256, show_label=False, tool=None, interactive=False)
+                fig1 = gr.Image(value=Image.open('assets/elevation.jpg'), type='pil', image_mode='RGB', height=256, show_label=False, interactive=False)
 
             with gr.Column(scale=0.8):
-                input_block = gr.ImageEditor(type='pil', image_mode='RGBA', label="Input to HarmonyView", height=256, tool=None, interactive=False)
+                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)
@@ -238,7 +238,7 @@ def run_demo():
                     seed = gr.Number(6033, label='Random seed', interactive=True)
                 run_btn = gr.Button('Run generation', variant='primary', interactive=True)
 
-        output_block = gr.ImageEditor(type='pil', image_mode='RGB', label="Outputs of HarmonyView", height=256, interactive=False)
+        output_block = gr.Image(type='pil', image_mode='RGB', label="Outputs of HarmonyView", height=256, interactive=False)
 
         image_block.change(fn=partial(sam_predict, mask_predictor, removal), inputs=[image_block], outputs=[sam_block], queue=True) \
             .success(fn=resize_inputs, inputs=[image_block, sam_block, crop_size, background_removal], outputs=[input_block], queue=True)