akhilchint/tiny-llama-api-endpoint

1from fastapi import FastAPI, Request2from fastapi.responses import HTMLResponse3 4app = FastAPI()5 6@app.get("/", response_class=HTMLResponse)7async def root():8    return """9    <h1>✅ Your FastAPI Space is working!</h1>10    <p>Try posting to <code>/generate</code> once you’ve wired in your model.</p>11    """12 13@app.post("/generate")14async def generate(request: Request):15    data = await request.json()16    prompt = data.get("prompt", "")17    return {"response": f"You said: {prompt}"}18