main.py

from fastapi import FastAPI
from app.routes import prediction
from app.routes.prediction import predict

from app.routes import home
import gradio as gr


app = FastAPI()


app.include_router(
    home.router,
    tags=["home"],
)


app.include_router(
    prediction.router,
    prefix="/predict",
    tags=["prediction"],
)

iface = gr.Interface(fn=predict, inputs="text", outputs="text")
iface.launch()