Update app.py

app.py

@@ -1,6 +1,6 @@
 import os
-os.system("pip uninstall -y gradio") 
-os.system('pip install gradio==4.0.0')
+# os.system("pip uninstall -y gradio") 
+# os.system('pip install gradio==4.0.0')
 #3.43.1
 import torch
 import torchvision
@@ -435,10 +435,10 @@ 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(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) 
                     with gr.Row():
                         sample = gr.Button("🎲 Sample New Model")
                         invert_button = gr.Button("⏪ Invert")