Create app.py

app.py

@@ -0,0 +1,45 @@
+from fastapi import FastAPI, UploadFile, File
+from transformers import AutoProcessor, AutoModelForVision2Seq
+from PIL import Image
+import torch
+import io
+
+app = FastAPI()
+
+MODEL_ID = "zai-org/GLM-OCR"
+
+print("Loading GLM-OCR model...")
+
+processor = AutoProcessor.from_pretrained(MODEL_ID)
+model = AutoModelForVision2Seq.from_pretrained(
+    MODEL_ID,
+    torch_dtype=torch.float32
+)
+
+@app.get("/")
+async def root():
+    return {"status": "GLM-OCR API is running"}
+
+@app.post("/ocr")
+async def extract_text(file: UploadFile = File(...)):
+    try:
+        contents = await file.read()
+        image = Image.open(io.BytesIO(contents)).convert("RGB")
+
+        inputs = processor(images=image, return_tensors="pt")
+
+        with torch.no_grad():
+            outputs = model.generate(**inputs, max_new_tokens=1024)
+
+        text = processor.batch_decode(outputs, skip_special_tokens=True)[0]
+
+        return {
+            "success": True,
+            "text": text
+        }
+
+    except Exception as e:
+        return {
+            "success": False,
+            "error": str(e)
+        }