gradio-sub-app / app.py
freddyaboulton's picture
use code
4b72be2
import gradio as gr
from fastapi import FastAPI
from fastapi.staticfiles import StaticFiles
import uvicorn
app = FastAPI()
@app.on_event("startup")
def print_foo():
print("FOO!!!")
def dream(prompt):
return ["static/lion.jpg"] * 2, {}
block = gr.Blocks().queue()
with block:
prompt = gr.Text()
gallery = gr.Gallery().style(grid=[2], height="auto")
contains_nfsw = gr.JSON(visible=False)
btn = gr.Button("Generate")
btn.click(dream, inputs=prompt, outputs=[gallery, contains_nfsw])
gradio_app = gr.routes.App.create_app(block)
app = gr.mount_gradio_app(app, gradio_app, "localhost", "7869", "/gradio")
app.mount("/", StaticFiles(directory="static", html=True), name="static")
uvicorn.run(app, host='0.0.0.0', port=7860)