amazonaws-la commited on
Commit
fa65ea1
·
verified ·
1 Parent(s): 8919369

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -72,7 +72,8 @@ def generate(
72
  pipe.load_lora_weights(lora)
73
  pipe.fuse_lora(lora_scale=0.7)
74
 
75
- init_image = Image.open("https://m.media-amazon.com/images/I/81zPcrN6m+L.jpg")
 
76
  init_image = init_image.resize((1024, 1024))
77
 
78
  if ENABLE_CPU_OFFLOAD:
 
72
  pipe.load_lora_weights(lora)
73
  pipe.fuse_lora(lora_scale=0.7)
74
 
75
+ response = requests.get("https://m.media-amazon.com/images/I/81zPcrN6m+L.jpg")
76
+ init_image = Image.open(BytesIO(response.content)
77
  init_image = init_image.resize((1024, 1024))
78
 
79
  if ENABLE_CPU_OFFLOAD: