post

main.py

@@ -1,8 +1,15 @@
 from fastapi import FastAPI
+from diffusers import StableDiffusionPipeline
 
 
 app = FastAPI()
 
 @app.get("/")
 def read_root():
-    return {"Hello": "World!"}
+    return {"Hello": "World!"}
+
+@app.post("/generate")
+def generate_image(input: str):
+    model = StableDiffusionPipeline.from_pretrained("runwayml/stable-diffusion-v1-5")
+
+    return {"image": "hello from the server", "input": input}