from fastapi import FastAPI
from fastapi.middleware.cors import CORSMiddleware
import gradio as gr

def image_generator(prompt):
    pass

interface = gr.Interface(fn=image_generator, inputs="text", outputs="image")
app = FastAPI()

app.add_middleware(
    CORSMiddleware,
    allow_origins=["*"],  
    allow_credentials=True,
    allow_methods=["*"],
    allow_headers=["*"],
)

app = gr.mount_gradio_app(app, interface, path="/")