Spaces:
Sleeping
Sleeping
File size: 359 Bytes
2579c7e 6dd78dc cf13da4 2579c7e c303997 2579c7e c303997 2579c7e 6dd78dc |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 |
from fastapi import FastAPI
from diffusers import StableDiffusionPipeline
app = FastAPI()
@app.get("/")
def read_root():
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} |