Update app.py

app.py

@@ -5,7 +5,7 @@ from PIL import Image
 from diffusers.models import AutoencoderKL
 from diffusers import AutoPipelineForInpainting, UNet2DConditionModel, DiffusionPipeline
 import diffusers
-from share_btn import community_icon_html, loading_icon_html, share_js
+
 
 
 vae = AutoencoderKL.from_pretrained("stabilityai/sd-vae-ft-mse")
@@ -91,17 +91,7 @@ css = '''
         transform: rotate(360deg);
     }
 }
-#share-btn-container {padding-left: 0.5rem !important; padding-right: 0.5rem !important; background-color: #000000; justify-content: center; align-items: center; border-radius: 9999px !important; max-width: 13rem; margin-left: auto;}
-div#share-btn-container > div {flex-direction: row;background: black;align-items: center}
-#share-btn-container:hover {background-color: #060606}
-#share-btn {all: initial; color: #ffffff;font-weight: 600; cursor:pointer; font-family: 'IBM Plex Sans', sans-serif; margin-left: 0.5rem !important; padding-top: 0.5rem !important; padding-bottom: 0.5rem !important;right:0;}
-#share-btn * {all: unset}
-#share-btn-container div:nth-child(-n+2){width: auto !important;min-height: 0px !important;}
-#share-btn-container .wrap {display: none !important}
-#share-btn-container.hidden {display: none!important}
 #prompt input{width: calc(100% - 160px);border-top-right-radius: 0px;border-bottom-right-radius: 0px;}
-#run_button{position:absolute;margin-top: 11px;right: 0;margin-right: 0.8em;border-bottom-left-radius: 0px;
-    border-top-left-radius: 0px;}
 #prompt-container{margin-top:-18px;}
 #prompt-container .form{border-top-left-radius: 0;border-top-right-radius: 0}
 #image_upload{border-bottom-left-radius: 0px;border-bottom-right-radius: 0px}
@@ -116,6 +106,7 @@ with image_blocks as demo:
                     with gr.Row(elem_id="prompt-container",equal_height=True):
                         with gr.Row():
                             prompt = gr.Textbox(placeholder="Your prompt", show_label=False, elem_id="prompt", lines=5, allow_resize=True)
+                        with gr.Row():
                             btn = gr.Button("Inpaint!", elem_id="run_button")
                     
                     with gr.Accordion(label="Advanced Settings", open=False):
@@ -124,21 +115,20 @@ with image_blocks as demo:
                             steps = gr.Number(value=40, minimum=10, maximum=100, step=1, label="steps")
                             strength = gr.Number(value=0.8, minimum=0.01, maximum=1.0, step=0.01, label="strength")
                             negative_prompt = gr.Textbox(label="negative_prompt", placeholder="Your negative prompt", info="what you don't want to see in the image")
+                        
+                        with gr.Row(equal_height=True):
+                            models = ["Realistic_V5.0","Realistic_V5.1","Realistic_V6.0","Epic_Realism"]
+                            model = gr.Dropdown(label="Models",choices=models,value="Realistic_V5.0")
                         with gr.Row(equal_height=True):
                             schedulers = ["DEISMultistepScheduler", "HeunDiscreteScheduler", "EulerDiscreteScheduler", "DPMSolverMultistepScheduler", "DPMSolverMultistepScheduler-Karras", "DPMSolverMultistepScheduler-Karras-SDE"]
                             scheduler = gr.Dropdown(label="Schedulers", choices=schedulers, value="DPMSolverMultistepScheduler-Karras")
                         
                 with gr.Column():
                     image_out = gr.Image(label="Output", elem_id="output-img", height=512, width=512)
-                    with gr.Group(elem_id="share-btn-container", visible=False) as share_btn_container:
-                        community_icon = gr.HTML(community_icon_html)
-                        loading_icon = gr.HTML(loading_icon_html)
-                        share_button = gr.Button("Share to community", elem_id="share-btn",visible=False)
             
 
-    btn.click(fn=predict, inputs=[image, prompt, negative_prompt, guidance_scale, steps, strength, scheduler], outputs=[image_out, share_btn_container], api_name='run')
-    prompt.submit(fn=predict, inputs=[image, prompt, negative_prompt, guidance_scale, steps, strength, scheduler], outputs=[image_out, share_btn_container])
-    share_button.click(None, [], [], _js=share_js)
+    btn.click(fn=predict, inputs=[image, prompt, negative_prompt, guidance_scale, steps, strength, model, scheduler], outputs=[image_out], api_name='run')
+    prompt.submit(fn=predict, inputs=[image, prompt, negative_prompt, guidance_scale, steps, strength, model, scheduler], outputs=[image_out])
 
 
 image_blocks.queue(max_size=25,api_open=True).launch(show_api=True)