amazonaws-la commited on
Commit
3454bb2
·
verified ·
1 Parent(s): f48135c

Update app.py

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