Update app.py

app.py

@@ -17,6 +17,8 @@ from PIL import Image
 models = ["Model A", "Model B", "Model C"]
 vae = ["VAE A", "VAE B", "VAE C"]
 controlnet_types = ["Canny", "Depth", "Normal", "Pose"]
+schedulers = ["Euler", "LMS", "DDIM"]
+
 
 # Fungsi placeholder
 def load_model(selected_model):
@@ -314,6 +316,26 @@ with gr.Blocks(css= "style.css") as app:
         
         controlnet_button.click(fn=controlnet_process_func, inputs=[controlnet_image_input, controlnet_dropdown, model_dropdown, vae_dropdown], outputs=controlnet_output_image)
 
+    # Tab untuk Intpainting
+    with gr.Tab ("Inpainting"):
+        with gr.Row():
+            with gr.Column():
+                image = gr.Image(source='upload', tool='sketch', elem_id="image_upload", type="pil", label="Upload",height=400)
+                btn = gr.Button("Inpaint!", elem_id="run_button")
+                prompt = gr.Textbox(placeholder="Your prompt (what you want in place of what is erased)", show_label=False, elem_id="prompt")
+                negative_prompt = gr.Textbox(label="negative_prompt", placeholder="Your negative prompt", info="what you don't want to see in the image")
+                guidance_scale = gr.Number(value=7.5, minimum=1.0, maximum=20.0, step=0.1, label="guidance_scale")
+                steps = gr.Number(value=20, minimum=10, maximum=30, step=1, label="steps")
+                strength = gr.Number(value=0.99, minimum=0.01, maximum=1.0, step=0.01, label="strength")
+                scheduler = gr.Dropdown(label="Schedulers", choices=schedulers, value="EulerDiscreteScheduler")
+            with gr.Column():
+                image_out = gr.Image(label="Output", elem_id="output-img")
+                   
+    btn.click(fn=predict, inputs=[image, prompt, negative_prompt, guidance_scale, steps, strength, scheduler], outputs=[image_out, share_btn_container])
+    prompt.submit(fn=predict, inputs=[image, prompt, negative_prompt, guidance_scale, steps, strength, scheduler], outputs=[image_out, share_btn_container])
+    
+
+
     
     # Tab untuk Describe
     with gr.Tab("Describe"):