| |
| |
| |
| |
| |
|
|
| from fastapi import FastAPI |
| from inference import generate_response |
|
|
| app = FastAPI() |
|
|
|
|
| @app.post("/predict") |
| def predict(payload: dict): |
| """ |
| Expected JSON: |
| { |
| "prompt": "user message" |
| } |
| """ |
|
|
| prompt = payload.get("prompt", "").strip() |
|
|
| if not prompt: |
| return {"response": "Empty prompt received."} |
|
|
| response = generate_response(prompt) |
| return {"response": response} |
|
|
|
|
| @app.get("/") |
| def root(): |
| return {"status": "DocuMint Space is running"} |
|
|
| if __name__ == "__main__": |
| import uvicorn |
| uvicorn.run(app, host="0.0.0.0", port=7860) |
|
|
