Create app.py

app.py

@@ -0,0 +1,33 @@
+from fastapi import FastAPI, UploadFile, File
+from rapidocr_onnxruntime import RapidOCR
+from PIL import Image
+import io
+import numpy as np
+
+app = FastAPI()
+ocr = RapidOCR(cls=False)
+
+@app.get("/")
+def read_root():
+    return {"status": "OCR API is running. Use POST /predict"}
+
+@app.post("/predict")
+async def predict(file: UploadFile = File(...)):
+    try:
+        contents = await file.read()
+        image = Image.open(io.BytesIO(contents))
+        
+        # Оптимизируем размер прямо на сервере HF (там 16 ГБ ОЗУ, но скорость важна)
+        if image.width > 1024 or image.height > 1024:
+            image.thumbnail((1024, 1024))
+            
+        img_array = np.array(image)
+        results, _ = ocr(img_array)
+        
+        if not results:
+            return {"text": ""}
+            
+        text = " ".join([item[1] for item in results]).strip()
+        return {"text": text}
+    except Exception as e:
+        return {"error": str(e)}