Update app.py
Browse files
app.py
CHANGED
@@ -14,7 +14,7 @@ model_path = "/content/uberRealisticPornMerge_urpmv12.instruct-pix2pix.safetenso
|
|
14 |
def main():
|
15 |
# Load the SafeTensor model from Colab
|
16 |
safe_pipe = StableDiffusionInstructPix2PixPipeline.from_pretrained(model_path, torch_dtype=torch.float16, safety_checker=None).to("cuda")
|
17 |
-
example_image = Image.open("
|
18 |
|
19 |
def load_example(
|
20 |
steps: int,
|
|
|
14 |
def main():
|
15 |
# Load the SafeTensor model from Colab
|
16 |
safe_pipe = StableDiffusionInstructPix2PixPipeline.from_pretrained(model_path, torch_dtype=torch.float16, safety_checker=None).to("cuda")
|
17 |
+
example_image = Image.open("/content/T/imgs_example.jpg").convert("RGB")
|
18 |
|
19 |
def load_example(
|
20 |
steps: int,
|