rootglitch commited on
Commit
da4aebc
·
1 Parent(s): 1f74d1c

Modified pil to open web file

Browse files
Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -11,6 +11,7 @@ import base64
11
  import numpy as np
12
  import math
13
  import scipy
 
14
  import torch
15
  import torchvision
16
  import gradio as gr
@@ -462,7 +463,9 @@ def generate_ai_bg(input_img, prompt):
462
 
463
  relight_img_path = result['images'][0]['url']
464
 
465
- relight_img = Image.open(relight_img_path).convert("RGBA")
 
 
466
 
467
  return relight_img
468
 
 
11
  import numpy as np
12
  import math
13
  import scipy
14
+ import requests
15
  import torch
16
  import torchvision
17
  import gradio as gr
 
463
 
464
  relight_img_path = result['images'][0]['url']
465
 
466
+ response = requests.get(relight_img_path, stream=True)
467
+
468
+ relight_img = Image.open(BytesIO(response.content)).convert("RGBA")
469
 
470
  return relight_img
471