asoderznik
commited on
Commit
•
ff2ef5c
1
Parent(s):
732c9a5
Update handler.py
Browse files- handler.py +2 -2
handler.py
CHANGED
@@ -46,8 +46,8 @@ class EndpointHandler():
|
|
46 |
image = Image.open(image_bytes)
|
47 |
prompt = inputs['prompt']
|
48 |
logger.info('image')
|
49 |
-
|
50 |
-
|
51 |
|
52 |
buffered = BytesIO()
|
53 |
upscaled_image.save(buffered, format="JPEG")
|
|
|
46 |
image = Image.open(image_bytes)
|
47 |
prompt = inputs['prompt']
|
48 |
logger.info('image')
|
49 |
+
|
50 |
+
upscaled_image = self.pipe(prompt, image).images[0]
|
51 |
|
52 |
buffered = BytesIO()
|
53 |
upscaled_image.save(buffered, format="JPEG")
|