Update app.py

app.py

@@ -438,13 +438,12 @@ with gr.Blocks(css="style.css") as demo:
     with gr.Column():
             with gr.Row():
                 with gr.Column():
-                    # input_image = gr.Image(source='upload', elem_id="image_upload", tool='sketch', type='pil', label="Upload image and draw to define mask",
-                    #                         height=512, width=512, brush_color='#00FFFF', mask_opacity=0.6)
-                    input_image = gr.ImageEditor(elem_id="image_upload", type='pil', label="Upload image and draw to define mask",
-                                            height=512, width=512, brush=gr.Brush(), layers=False) 
+                    input_image = gr.Image(label="Identity from Sampled or Uploaded Model", height=512, width=512)
+                    # input_image = gr.ImageEditor(elem_id="image_upload", type='pil', label="Upload image and draw to define mask",
+                    #                         height=512, width=512, brush=gr.Brush(), layers=False) 
                     with gr.Row():
                         sample = gr.Button("🎲 Sample New Model")
-                        invert_button = gr.Button("⏪ Invert")
+                        # invert_button = gr.Button("⏪ Invert")
                 with gr.Column():
                     image_slider = ImageSlider(position=0.5, type="pil", height=512, width=512)
                     # gallery1 = gr.Image(label="Identity from Original Model",height=512, width=512, interactive=False)
@@ -517,9 +516,9 @@ with gr.Blocks(css="style.css") as demo:
 
 
 
-        invert_button.click(fn=run_inversion,
-                    inputs=[input_image, pcs, epochs, weight_decay,lr], 
-                    outputs = [image_slider, file_output])
+        # invert_button.click(fn=run_inversion,
+        #             inputs=[input_image, pcs, epochs, weight_decay,lr], 
+        #             outputs = [image_slider, file_output])
         
             
         sample.click(fn=sample_then_run, outputs=[input_image, file_output])