Spaces:
Running
on
Zero
Running
on
Zero
Commit
·
da4aebc
1
Parent(s):
1f74d1c
Modified pil to open web file
Browse files
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 |
-
|
|
|
|
|
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 |
|