Spaces:
Running
on
Zero
Running
on
Zero
amazonaws-la
commited on
Commit
•
8ac809c
1
Parent(s):
a5eaa6f
Update app.py
Browse files
app.py
CHANGED
@@ -57,7 +57,6 @@ def generate(
|
|
57 |
model = 'SG161222/Realistic_Vision_V6.0_B1_noVAE',
|
58 |
vaecall = 'stabilityai/sd-vae-ft-mse',
|
59 |
lora = 'amazonaws-la/juliette',
|
60 |
-
url = "https://raw.githubusercontent.com/CompVis/stable-diffusion/main/assets/stable-samples/img2img/sketch-mountains-input.jpg"
|
61 |
lora_scale: float = 0.7,
|
62 |
) -> PIL.Image.Image:
|
63 |
if torch.cuda.is_available():
|
@@ -73,6 +72,8 @@ def generate(
|
|
73 |
pipe.load_lora_weights(lora)
|
74 |
pipe.fuse_lora(lora_scale=0.7)
|
75 |
|
|
|
|
|
76 |
response = requests.get(url)
|
77 |
init_image = Image.open(BytesIO(response.content)).convert("RGB")
|
78 |
init_image = init_image.resize((1024, 1024))
|
|
|
57 |
model = 'SG161222/Realistic_Vision_V6.0_B1_noVAE',
|
58 |
vaecall = 'stabilityai/sd-vae-ft-mse',
|
59 |
lora = 'amazonaws-la/juliette',
|
|
|
60 |
lora_scale: float = 0.7,
|
61 |
) -> PIL.Image.Image:
|
62 |
if torch.cuda.is_available():
|
|
|
72 |
pipe.load_lora_weights(lora)
|
73 |
pipe.fuse_lora(lora_scale=0.7)
|
74 |
|
75 |
+
url = "https://raw.githubusercontent.com/CompVis/stable-diffusion/main/assets/stable-samples/img2img/sketch-mountains-input.jpg"
|
76 |
+
|
77 |
response = requests.get(url)
|
78 |
init_image = Image.open(BytesIO(response.content)).convert("RGB")
|
79 |
init_image = init_image.resize((1024, 1024))
|