3morrrrr's picture
Deploy handwriting synthesis API
e718762
raw
history blame
673 Bytes
from fastapi import FastAPI
from fastapi.middleware.cors import CORSMiddleware
from pydantic import BaseModel
import numpy as np
import tensorflow as tf
app = FastAPI()
app.add_middleware(
CORSMiddleware,
allow_origins=["*"],
allow_methods=["*"],
allow_headers=["*"],
)
class TextInput(BaseModel):
text: str
style: int = 0
@app.post("/generate")
async def generate_handwriting(input: TextInput):
# Load model and generate handwriting
# Placeholder implementation
return {"status": "success", "message": "Handwriting generation endpoint"}
if __name__ == "__main__":
import uvicorn
uvicorn.run(app, host="0.0.0.0", port=7860)