main.py

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

app = FastAPI()

origins = ["https://lyrickle.vercel.app", "http://lyrickle.vercel.app"]

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


taylor = pipeline("text-generation", model="BhavyaMuni/model-generator")

@app.get("/")
def read_root():
    return {"Hello": "World"}

@app.get("/taylor-swift")
def get_taylor_swift(query: str):
    predict = taylor(query, max_length=100, do_sample=True, temperature=0.7)
    return {"output": predict[0]['generated_text']}