bp7274/UnityAPI

1from fastapi import FastAPI, HTTPException2from pydantic import BaseModel3from transformers import pipeline4# Load model5generator = pipeline("text-generation", model="bp7274/soulprint-model")6app = FastAPI()7class InputData(BaseModel):8    age: int9    location: str10    profession: str11    personality: str12@app.get("/")13def health():14    return {"status": "ready"}15 16@app.post("/generate")17def generate_quote(data: InputData):18    try:19        prompt = f"{data.age}, {data.location}, {data.profession}, {data.personality}."20        result = generator(21            prompt,22            max_length=120,23            temperature=0.9,24            top_p=0.95,25            repetition_penalty=1.2,26            do_sample=True27        )[0]["generated_text"]28        # Strip only the prompt portion (no fallback check)29        quote = result[len(prompt):].strip()30        return {"quote": quote}31    except Exception as e:32        raise HTTPException(status_code=500, detail=str(e))