HarmonyView update

.idea/workspace.xml

@@ -4,10 +4,7 @@
     <option name="autoReloadType" value="SELECTIVE" />
   </component>
   <component name="ChangeListManager">
-    <list default="true" id="a993d736-6297-4164-9c29-6b2ab1055a96" name="변경" comment="HarmonyView update">
-      <change beforePath="$PROJECT_DIR$/.idea/workspace.xml" beforeDir="false" afterPath="$PROJECT_DIR$/.idea/workspace.xml" afterDir="false" />
-      <change beforePath="$PROJECT_DIR$/app.py" beforeDir="false" afterPath="$PROJECT_DIR$/app.py" afterDir="false" />
-    </list>
+    <list default="true" id="a993d736-6297-4164-9c29-6b2ab1055a96" name="변경" comment="HarmonyView update" />
     <option name="SHOW_DIALOG" value="false" />
     <option name="HIGHLIGHT_CONFLICTS" value="true" />
     <option name="HIGHLIGHT_NON_ACTIVE_CHANGELIST" value="false" />
@@ -65,7 +62,7 @@
       <workItem from="1704106389325" duration="3000" />
       <workItem from="1707152826665" duration="2000" />
       <workItem from="1709009070011" duration="10000" />
-      <workItem from="1717834285162" duration="735000" />
+      <workItem from="1717834285162" duration="1496000" />
     </task>
     <task id="LOCAL-00001" summary="error resolve">
       <option name="closed" value="true" />
@@ -259,7 +256,15 @@
       <option name="project" value="LOCAL" />
       <updated>1717834384940</updated>
     </task>
-    <option name="localTasksCounter" value="25" />
+    <task id="LOCAL-00025" summary="HarmonyView update">
+      <option name="closed" value="true" />
+      <created>1717835213390</created>
+      <option name="number" value="00025" />
+      <option name="presentableId" value="LOCAL-00025" />
+      <option name="project" value="LOCAL" />
+      <updated>1717835213390</updated>
+    </task>
+    <option name="localTasksCounter" value="26" />
     <servers />
   </component>
   <component name="Vcs.Log.Tabs.Properties">

README.md

@@ -4,7 +4,7 @@ emoji: 🚀
 colorFrom: indigo
 colorTo: pink
 sdk: gradio
-sdk_version: 3.50.2
+sdk_version: 4.35.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.Image(type='pil', image_mode='RGBA', height=256, label='Input image', interactive=True)
+    image_block = gr.ImageEditor(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.Image(value=Image.open('assets/crop_size.jpg'), type='pil', image_mode='RGB', height=256, show_label=False, tool=None, interactive=False)
+                fig0 = gr.ImageEditor(value=Image.open('assets/crop_size.jpg'), type='pil', image_mode='RGB', height=256, show_label=False, tool=None, interactive=False)
 
 
             with gr.Column(scale=0.8):
-                sam_block = gr.Image(type='pil', image_mode='RGBA', label="SAM output", height=256, interactive=False)
+                sam_block = gr.ImageEditor(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.Image(value=Image.open('assets/elevation.jpg'), type='pil', image_mode='RGB', height=256, show_label=False, tool=None, interactive=False)
+                fig1 = gr.ImageEditor(value=Image.open('assets/elevation.jpg'), type='pil', image_mode='RGB', height=256, show_label=False, tool=None, interactive=False)
 
             with gr.Column(scale=0.8):
-                input_block = gr.Image(type='pil', image_mode='RGBA', label="Input to HarmonyView", height=256, tool=None, interactive=False)
+                input_block = gr.ImageEditor(type='pil', image_mode='RGBA', label="Input to HarmonyView", height=256, tool=None, 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.Image(type='pil', image_mode='RGB', label="Outputs of HarmonyView", height=256, interactive=False)
+        output_block = gr.ImageEditor(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)