asoderznik
commited on
Commit
•
26fde58
1
Parent(s):
58dfe8e
Update handler.py
Browse files- handler.py +16 -10
handler.py
CHANGED
@@ -5,6 +5,12 @@ from torch import autocast
|
|
5 |
from diffusers import StableDiffusionPipeline
|
6 |
import base64
|
7 |
from io import BytesIO
|
|
|
|
|
|
|
|
|
|
|
|
|
8 |
|
9 |
|
10 |
# set device
|
@@ -28,19 +34,19 @@ class EndpointHandler():
|
|
28 |
A :obj:`dict`:. base64 encoded image
|
29 |
"""
|
30 |
inputs = data.pop("inputs", data)
|
31 |
-
|
32 |
-
|
33 |
-
|
34 |
|
35 |
-
|
36 |
-
|
37 |
-
|
38 |
|
39 |
# decode base64 image to PIL
|
40 |
-
|
41 |
-
|
42 |
-
|
43 |
-
|
44 |
|
45 |
# run inference pipeline
|
46 |
#upscaled_image = self.pipe(prompt="", image = image).images[0]
|
|
|
5 |
from diffusers import StableDiffusionPipeline
|
6 |
import base64
|
7 |
from io import BytesIO
|
8 |
+
from transformers.utils import logging
|
9 |
+
|
10 |
+
logging.set_verbosity_info()
|
11 |
+
logger = logging.get_logger("transformers")
|
12 |
+
logger.info("INFO")
|
13 |
+
logger.warning("WARN")
|
14 |
|
15 |
|
16 |
# set device
|
|
|
34 |
A :obj:`dict`:. base64 encoded image
|
35 |
"""
|
36 |
inputs = data.pop("inputs", data)
|
37 |
+
logger.info("Printing inputs")
|
38 |
+
logger.info(inputs)
|
39 |
+
logger.info("")
|
40 |
|
41 |
+
logger.info("Printing image")
|
42 |
+
logger.info(inputs['image'])
|
43 |
+
logger.info("")
|
44 |
|
45 |
# decode base64 image to PIL
|
46 |
+
image = Image.open(BytesIO(base64.b64decode(inputs['image'])))
|
47 |
+
logger.info("Printing loaded image into library")
|
48 |
+
logger.info(image)
|
49 |
+
logger.info("")
|
50 |
|
51 |
# run inference pipeline
|
52 |
#upscaled_image = self.pipe(prompt="", image = image).images[0]
|