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 |
|