Update app.py

app.py

@@ -36,6 +36,17 @@ def controlnet_process(image, controlnet_type, model):
     # Logika untuk memproses gambar menggunakan ControlNet
     return f"Proses gambar dengan ControlNet '{controlnet_type}' dan model '{model}'"
 
+
+def controlnet_process_func(image, controlnet_type, model):
+    # Update fungsi sesuai kebutuhan
+    return controlnet_process(image, controlnet_type, model)
+
+def intpaint_func (image, controlnet_type, model):
+    # Update fungsi sesuai kebutuhan
+    return controlnet_process(image, controlnet_type, model)
+        
+
+
 #wd tagger
 
 # Dataset v3 series of models:
@@ -309,11 +320,6 @@ with gr.Blocks(css= "style.css") as app:
             
             with gr.Column():
                 controlnet_output_image = gr.Image(label="Hasil ControlNet")
-        
-        def controlnet_process_func(image, controlnet_type, model):
-            # Update fungsi sesuai kebutuhan
-            return controlnet_process(image, controlnet_type, model)
-        
         controlnet_button.click(fn=controlnet_process_func, inputs=[controlnet_image_input, controlnet_dropdown, model_dropdown, vae_dropdown], outputs=controlnet_output_image)
 
     # Tab untuk Intpainting
@@ -331,7 +337,7 @@ with gr.Blocks(css= "style.css") as app:
             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])
+    btn.click(fn=intpaint_func, inputs=[image, prompt, negative_prompt, guidance_scale, steps, strength, scheduler], outputs=[image_out, share_btn_container])