amazonaws-la commited on
Commit
f48135c
·
verified ·
1 Parent(s): 244b495

Update app.py

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