gradio-subapp / app.py
freddyaboulton's picture
Fix
d978202
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 ["images/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("/img", StaticFiles(directory="images", html=True), name="images")
uvicorn.run(app, port=7860)