Update CaesarAIART/caesaraiart.py

CaesarAIART/caesaraiart.py

@@ -27,10 +27,12 @@ class CaesarAIART:
         image_bytes = query({"inputs": prompt})
 
         image = Image.open(io.BytesIO(image_bytes))
-        filtered_image = io.BytesIO()
-        image.save(filtered_image, "PNG")
-        filtered_image.seek(0)
-        return filtered_image
+          # create a thumbnail image
+        imgio = io.BytesIO()
+        image.save(imgio, 'JPEG')
+        imgio.seek(0)
+        return imgio
+        #return StreamingResponse(content=imgio, media_type="image/jpeg"
 
 if __name__ == "__main__":
     prompt = "a photo of an astronaut riding a horse on mars"