CaesarAI Handwriting

CaesarHandwriting/caesarhandwriting.py

@@ -7,7 +7,7 @@ class CaesarHandWriting:
     def __init__(self) -> None:
         self.processor = TrOCRProcessor.from_pretrained('microsoft/trocr-large-handwritten')
         self.model = VisionEncoderDecoderModel.from_pretrained('microsoft/trocr-large-handwritten')
-    def translate(self,url='https://fki.tic.heia-fr.ch/static/img/a01-122-02-00.jpg'):
+    def translate(self,url):
         image = Image.open(requests.get(url, stream=True).raw).convert("RGB")
         pixel_values = self.processor(images=image, return_tensors="pt").pixel_values
         generated_ids = self.model.generate(pixel_values)

main.py

@@ -57,9 +57,10 @@ def caesaraihome():
 def caesarailogo():
     return FileResponse(f"{CURRENT_DIR}/CaesarAILogo.png")
 
-@app.get("/caesarhandwriting")
-def caesaraihandwriting(url: Union[str,None]):
-    text = caesarhandwriting.translate(url)
+@app.post("/caesarhandwriting")
+def caesaraihandwriting(urljson : CaesarAIHandwritingModel):
+    urljson = dict(urljson)
+    text = caesarhandwriting.translate(urljson["url"])
     return {"message":text}