Update app.py

app.py

@@ -4,10 +4,10 @@ from PIL import Image
 from diffusers import StableDiffusionInstructPix2PixPipeline
 
 # Define the path to the SafeTensor model
-model_path = "/content/uberRealisticPornMerge_urpmv12.instruct-pix2pix.safetensors"
+model_path = "/content/UberRealisticPornMerge_urpmv12.instruct-pix2pix.safetensors"
 
 # Load the SafeTensor model
-safe_pipe = StableDiffusionInstructPix2PixPipeline.from_pretrained(model_path, torch_dtype=torch.float16, safety_checker=None).to("cuda")
+safe_pipe = torch.load(model_path, map_location=torch.device('cuda'))
 
 def generate_edited_image(input_image):
     # Convert the Gradio Image object to a PIL Image