Update TextGen/router.py
Browse files- TextGen/router.py +1 -1
TextGen/router.py
CHANGED
@@ -245,4 +245,4 @@ def Imagen(image:ImageGen=None):
|
|
245 |
img_byte_arr = img_byte_arr.getvalue()
|
246 |
|
247 |
# Return the image as a PNG response
|
248 |
-
return Response(content=
|
|
|
245 |
img_byte_arr = img_byte_arr.getvalue()
|
246 |
|
247 |
# Return the image as a PNG response
|
248 |
+
return Response(content=img_byte_arr, media_type="image/png")
|