Spaces:

Jofthomas
/

evBackend

Running

Jofthomas HF staff commited on Aug 9

Commit

694028b

•

1 Parent(s): 923b7bc

Update TextGen/router.py

Files changed (1) hide show

TextGen/router.py CHANGED Viewed

@@ -234,6 +234,6 @@ async def generate_song():
     return infos
 @app.post('/generate_image')
-def Imagen(prompt:ImageGen=None):
-    image_bytes=generate_image(ImageGen.prompt)
     return Response(content=image_bytes, media_type="image/png")

     return infos
 @app.post('/generate_image')
+def Imagen(image:ImageGen=None):
+    image_bytes=generate_image(image.prompt)
     return Response(content=image_bytes, media_type="image/png")