Update app.py

app.py

@@ -14,7 +14,7 @@ from matplotlib import pyplot as plt
 from torchvision import transforms
 from diffusers import DiffusionPipeline
 
-from share_btn import community_icon_html, loading_icon_html, share_js
+# from share_btn import community_icon_html, loading_icon_html, share_js
 
 HF_TOKEN_SD = os.environ.get('HF_TOKEN_SD')  or True
 
@@ -101,13 +101,16 @@ with image_blocks as demo:
                         )
                 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, prompt], outputs=[image_out, community_icon, loading_icon, share_button])
+            btn.click(fn=predict, inputs=[image, prompt], outputs=[image_out]) #, community_icon, loading_icon, share_button])
             share_button.click(None, [], [], _js=share_js)