Update app.py

app.py

@@ -272,23 +272,22 @@ with image_blocks as demo:
                 with gr.Column():
                     image = gr.Image(source='upload', tool='sketch', elem_id="image_upload", type="pil", label="Upload") #.style(height=400)
                     with gr.Row(elem_id="prompt-container").style(mobile_collapse=False, equal_height=True):
-                        # prompt = gr.Textbox(placeholder = 'Your prompt (what you want in place of what is erased)', show_label=False, elem_id="input-text")
+                        prompt = gr.Textbox(placeholder = 'Your prompt (what you want in place of what is erased)', show_label=False, elem_id="input-text")
                         btn = gr.Button("Done!").style(
                             margin=True,
                             rounded=(True, True, True, True),
                             full_width=True,
-                        )
-                '''
+                        )                
                 with gr.Column():
                     image_out = gr.Image(label="Output", elem_id="output-img").style(height=400)
                     with gr.Group(elem_id="share-btn-container"):
                         community_icon = gr.HTML(community_icon_html, visible=False)
                         loading_icon = gr.HTML(loading_icon_html, visible=False)
                         share_button = gr.Button("Share to community", elem_id="share-btn", visible=False)
-                '''
+                
 
-            # btn.click(fn=predict, inputs=[image, prompt], outputs=[image_out, community_icon, loading_icon, share_button])
-            btn.click(fn=predict, inputs=[image], outputs=[image]) #, community_icon, loading_icon, share_button])
-            # share_button.click(None, [], [], _js=share_js)
+            btn.click(fn=predict, inputs=[image, prompt], outputs=[image_out, community_icon, loading_icon, share_button])
+            #btn.click(fn=predict, inputs=[image], outputs=[image]) #, community_icon, loading_icon, share_button])
+            share_button.click(None, [], [], _js=share_js)
 
 image_blocks.launch()