add fastapi

main.py

@@ -1,6 +1,13 @@
+from fastapi import FastAPI
 import gradio as gr
 import torch
 
+app = FastAPI()
+
+@app.get("/")
+def read_root():
+    return {"Hello": "World"}
+
 def generate_image(prompt, negative_prompt, seed, guidance_scale, num_inference_steps):
     return torch.rand(512, 512, 3)
 
@@ -112,4 +119,8 @@ with gr.Blocks(css=css) as demo:
         outputs = [result]
     )
 
-demo.queue().launch()
+app = gr.mount_gradio_app(app, demo, "/demo")
+
+if __name__ == "__main__":
+    import uvicorn
+    uvicorn.run(app, host="0.0.0.0", port=8000)