Update app.py

app.py

@@ -1,6 +1,6 @@
 import os
-# os.system("pip uninstall -y gradio") 
-# #os.system('pip install gradio==3.43.1')
+os.system("pip uninstall -y gradio") 
+os.system('pip install gradio==3.43.1')
 
 import torch
 import torchvision
@@ -110,7 +110,6 @@ def inference(prompt, negative_prompt, guidance_scale, ddim_steps, seed):
 
 
 
-# @spaces.GPU()
 @torch.no_grad()
 def edit_inference(input_image, prompt, negative_prompt, guidance_scale, ddim_steps, seed, start_noise, a1, a2, a3, a4):
 
@@ -197,7 +196,7 @@ def edit_inference(input_image, prompt, negative_prompt, guidance_scale, ddim_st
 
     return (image, input_image["background"])
 
-# @spaces.GPU()
+
 def sample_then_run():
     sample_model()    
     prompt = "sks person"
@@ -342,7 +341,6 @@ def invert(image, mask, pcs=10000, epochs=400, weight_decay = 1e-10, lr=1e-1):
     return network
 
 
-# @spaces.GPU(duration=200)
 def run_inversion(input_image, pcs, epochs, weight_decay,lr):
     global network
     init_image = input_image["background"].convert("RGB").resize((512, 512))
@@ -364,7 +362,6 @@ def run_inversion(input_image, pcs, epochs, weight_decay,lr):
 
 
 
-# @spaces.GPU()
 def file_upload(file):
     global unet
     del unet
@@ -438,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")