Update app.py

app.py

@@ -13,9 +13,9 @@ from gradio_client import Client, handle_file
 
 # Load the model once globally to avoid repeated loading
 def load_inpainting_model():
-    #model_path = "uberRealisticPornMerge_v23Inpainting.safetensors"
-    #model_path = "pornmasterFantasy_v4-inpainting.safetensors"
-    model_path = "pornmasterAmateur_v6Vae-inpainting.safetensors"
+    model_path = "uberRealisticPornMerge_v23Inpainting.safetensors"
+    model_path = "pornmasterFantasy_v4-inpainting.safetensors"
+    #model_path = "pornmasterAmateur_v6Vae-inpainting.safetensors"
     device = "cpu"  # Explicitly use CPU
     pipe = StableDiffusionInpaintPipeline.from_single_file(
         model_path,
@@ -57,7 +57,7 @@ def inpaint_image(input_image, mask_image):
     prompt = "undress, naked"
     result = inpaint_pipeline(prompt=prompt, image=input_image, mask_image=mask_image)
     inpainted_image = result.images[0]
-    inpainted_image = resize_to_match(input_image, inpainted_image)
+    #inpainted_image = resize_to_match(input_image, inpainted_image)
     return inpainted_image
 
 # Function to process input image and mask