Jofthomas HF staff commited on
Commit
c3df33a
1 Parent(s): 360f4e7

Update TextGen/router.py

Browse files
Files changed (1) hide show
  1. TextGen/router.py +0 -1
TextGen/router.py CHANGED
@@ -245,5 +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=img_byte_arr, media_type="image/png")
249
  return Response(content=image_bytes, media_type="image/png")
 
245
  img_byte_arr = img_byte_arr.getvalue()
246
 
247
  # Return the image as a PNG response
 
248
  return Response(content=image_bytes, media_type="image/png")