main.py

from fastapi import FastAPI
from fastapi.middleware.cors import CORSMiddleware

app = FastAPI(title="Deploying FastAPI Apps on Huggingface")


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


@app.get("/", tags=["Home"])
def api_home():
    return {"detail": "Welcome to FastAPI TextGen Tutorial!"}


@app.post(
    "/api/generate",
    summary="Generate text from prompt",
    tags=["Generate"],
    response_model=Generate,
)
def inference(input_prompt: str):
    return "Hello"