update layout

app.py

@@ -5,7 +5,7 @@ import gradio as gr
 # First, running the inversion command to obtain the input noise that will reconstruct the image. 
 # It will save the inversion as output/test_cat/inversion/image-name.pt 
 # BLIP-generated caption prompt is saved as output/test_cat/prompt/image-name.txt - eg., a painting of a cat sitting on top of a ball
-def inversion(image_in, progress=gr.Progress(track_tqdm=True)):
+def inversion(image_in):  #, progress=gr.Progress(track_tqdm=True)):
   progress(0, desc="Starting...")
   # saving the input image
   image_in.save("assets/test_images/cats/input_image.png")
@@ -20,7 +20,7 @@ def inversion(image_in, progress=gr.Progress(track_tqdm=True)):
 
 # Performing image editing with the editing directions 
 # This will save the edited image as output/test_cat/edit/image-name.png
-def image_edit(task_name, progress=gr.Progress(track_tqdm=True)):
+def image_edit(task_name): #, progress=gr.Progress(track_tqdm=True)):
   progress(0, desc="Starting...")
   # Run the script file
   subprocess.run(["python", "src/edit_real.py", "--inversion", "output/test_cat/inversion/input_image.pt", 
@@ -30,7 +30,7 @@ def image_edit(task_name, progress=gr.Progress(track_tqdm=True)):
 
 
 #Similarly, we can edit the synthetic images generated by Stable Diffusion with the following command.
-def synthetic_image_edit(prompt, task_name, progress=gr.Progress(track_tqdm=True)):
+def synthetic_image_edit(prompt, task_name): #, progress=gr.Progress(track_tqdm=True)):
   progress(0, desc="Starting...")
   # Run the script file
   subprocess.run(["python", "src/edit_synthetic.py", "--prompt_str", prompt, 
@@ -64,7 +64,7 @@ with gr.Blocks() as demo:
                 Please visit their website and <a href="https://github.com/pix2pixzero/pix2pix-zero" target="_blank">github repo</a> for more details.
                 </p></div>""")
   with gr.Row():
-    image_in = gr.Image(type="pil", label="Upload the image of a cat or a dog you want to translate")
+    image_in = gr.Image(type="pil", label="Start by uploading an image of a Cat or a Dog that you want to translate")
     with gr.Column():
       btn_inversion = gr.Button("Get input noise",visible=False )
       with gr.Row():
@@ -72,10 +72,10 @@ with gr.Blocks() as demo:
         inversion_file = gr.File(visible=False)
         #task_name = gr.Textbox()
   with gr.Row():
-    image_out = gr.Image(visible=False, label="Trnaslated Image output")
+    image_out = gr.Image(visible=False, label="Translated Image output")
     with gr.Column():
       task_name_radio  = gr.Radio(choices = ["cat2dog", "dog2cat",], type="value", visible=False, label="Select a task that you want to accomplish") #, value="cat2dog"),
-      btn_imageedit = gr.Button(value="Edit the image!",visible=False)
+      btn_imageedit = gr.Button(value="Translate the image!",visible=False)
       html_tag = gr.HTML(value="""<h3 style="font-weight: 900; margin-bottom: 7px; margin-top: 5px;">
               Generate synthetic images using Stable Diffusion and Translate them on the fly using Pix2Pix-Zero</h3>""", visible=False)
       prompt_synth = gr.Textbox(visible=False, label="Type in a prompt to generate an Image using SD")