Working on it

routers/training.py

@@ -18,4 +18,5 @@ router = APIRouter()
 async def hifunction():
     prompt = "a photograph of an astronaut riding a horse"
     image = pipe(prompt).images[0]
-    return [image]
+    print(image)
+    return ["done"]