Jofthomas HF staff commited on
Commit
15798be
1 Parent(s): c3df33a

Update TextGen/router.py

Browse files
Files changed (1) hide show
  1. 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=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=img_byte_arr, media_type="image/png")