amazonaws-la commited on
Commit
d4c7bff
·
verified ·
1 Parent(s): deaadab

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -58,6 +58,7 @@ def generate(
58
  model = 'SG161222/Realistic_Vision_V6.0_B1_noVAE',
59
  vaecall = 'stabilityai/sd-vae-ft-mse',
60
  lora = 'amazonaws-la/juliette',
 
61
  lora_scale: float = 0.7,
62
  ):
63
  if torch.cuda.is_available():
@@ -73,7 +74,6 @@ def generate(
73
  pipe.load_lora_weights(lora)
74
  pipe.fuse_lora(lora_scale=0.7)
75
 
76
- url = "https://raw.githubusercontent.com/CompVis/stable-diffusion/main/assets/stable-samples/img2img/sketch-mountains-input.jpg"
77
  response = requests.get(url)
78
  init_image = Image.open(BytesIO(response.content)).convert("RGB")
79
  init_image = init_image.resize((1024, 1024))
 
58
  model = 'SG161222/Realistic_Vision_V6.0_B1_noVAE',
59
  vaecall = 'stabilityai/sd-vae-ft-mse',
60
  lora = 'amazonaws-la/juliette',
61
+ url = "https://m.media-amazon.com/images/I/81zPcrN6m+L.jpg"
62
  lora_scale: float = 0.7,
63
  ):
64
  if torch.cuda.is_available():
 
74
  pipe.load_lora_weights(lora)
75
  pipe.fuse_lora(lora_scale=0.7)
76
 
 
77
  response = requests.get(url)
78
  init_image = Image.open(BytesIO(response.content)).convert("RGB")
79
  init_image = init_image.resize((1024, 1024))