benkada/IAAimages

1from fastapi import FastAPI2from fastapi.staticfiles import StaticFiles3from fastapi.responses import FileResponse4from transformers import pipeline5 6app = FastAPI()7 8# Mount static files9app.mount("/static", StaticFiles(directory="static"), name="static")10 11# Initialize the pipeline with a simpler model12pipe = pipeline("text-generation", model="gpt2")13 14@app.get("/")15async def home():16    return FileResponse("index.html")17 18@app.get("/generate")19def generate(text: str):20    output = pipe(text, max_length=100)21    return {"output": output[0]['generated_text']}22