File size: 622 Bytes
bde4812 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 |
from fastapi import FastAPI
from pydantic import BaseModel
from happytransformer import HappyTextToText, TTSettings
app = FastAPI()
happy_tt = HappyTextToText("T5", "vennify/t5-base-grammar-correction")
class InputText(BaseModel):
txt: str
@app.post("/correct_grammar/")
def correct_grammar(input_text: InputText):
args = TTSettings(num_beams=5, min_length=1, max_length=100000)
corrected_text = happy_tt.generate_text(f"grammar: {input_text.txt}", args=args)
return {"corrected_text": corrected_text.text}
if __name__ == "__main__":
import uvicorn
uvicorn.run(app, host="0.0.0.0", port=8000)
|